프로그래머스 문제 (Java) 추억 점수

문제 이름 :

추억 점수

 

문제 설명:

사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매기려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["may", "kein", "kain"]이고 각 인물의 그리움 점수가 [5점, 10점, 1점] 일 때 해당 사진의 추억 점수는 16(5 + 10 + 1)점이 됩니다. 다른 사진 속 인물의 이름이 ["kali", "mari", "don", "tony"]이고 ["kali", "mari", "don"]의 그리움 점수가 각각 [11점, 1점, 55점]]이고, "tony"는 그리움 점수가 없을 때, 이 사진의 추억 점수는 3명의 그리움 점수를 합한 67(11 + 1 + 55)점입니다.

그리워하는 사람의 이름을 담은 문자열 배열 name, 각 사람별 그리움 점수를 담은 정수 배열 yearning, 각 사진에 찍힌 인물의 이름을 담은 이차원 문자열 배열 photo가 매개변수로 주어질 때, 사진들의 추억 점수를 photo에 주어진 순서대로 배열에 담아 return하는 solution 함수를 완성해 주세요.

제한사항:
3 ≤ name의 길이 = yearning의 길이≤ 100
3 ≤ name의 원소의 길이 ≤ 7
name의 원소들은 알파벳 소문자로만 이루어져 있습니다.
name에는 중복된 값이 들어가지 않습니다.
1 ≤ yearning[i] ≤ 100
yearning [i]는 i번째 사람의 그리움 점수입니다.
3 ≤ photo의 길이 ≤ 100
1 ≤ photo[i]의 길이 ≤ 100
3 ≤ photo[i]의 원소(문자열)의 길이 ≤ 7
photo [i]의 원소들은 알파벳 소문자로만 이루어져 있습니다.
photo [i]의 원소들은 중복된 값이 들어가지 않습니다.

 

예시 배열::

name

["may", "kein", "kain", "radi"]   

 

yearning

[5, 10, 1, 3]

 

pohoto

[["may", "kein", "kain", "radi"], ["may", "kein", "brin", "deny"], ["kon", "kain", "may", "coni"]]

 

코드:

import java.util.*;

class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        int[] answer = new int[photo.length];
        
        Map<String, Integer> map = new HashMap<>();
        
        for(int i = 0; i < name.length; i++){
            map.put(name[i],yearning[i]);
        }
        
        for(int i = 0; i < photo.length; i++){
            for(int j = 0; j < photo[i].length; j++){
                if(map.containsKey(photo[i][j])){
                    answer[i] += map.get(photo[i][j]);
                }else {
                    answer[i] += 0;
                }
            }
        }
        
        return answer;
    }
}

 

Map의 사용이유:

photo 배열을 보면 name순서대로 있는 것도 있지만 그렇지 않은 것도 있어서 

배열 그대로를 사용하게 된다면 코드가 복잡해 질 것 같아서 Map을 사용하였습니다.