[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 함수를 완성해주세요.

 

코딩테스트를 처음 해볼 땐 어떤 자료구조를 사용할지 막막했는데 몇 번 풀다보니 HashMap과 친해져 가는 것 같다.

이번 풀이에서는 맵에 없는 키값이 들어올 경우를 대비해 getOrDefault() 를 사용했다.

 

문제풀이

import java.util.HashMap;

class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        int[] answer = new int[photo.length];
        HashMap<String, Integer> nameMap = new HashMap<String, Integer>();
        
        //name배열을 찾기 쉽게 HashMap에 저장
        for (int i=0; i<name.length; i++)
            nameMap.put(name[i], yearning[i]);
        
        //photo배열의 값 더하기
        for(int i=0; i<photo.length; i++)  {
            for (int j=0; j<photo[i].length; j++) {
                answer[i] += nameMap.getOrDefault(photo[i][j],0);
            }
        }
        return answer;
    }
}