[프로그래머스] 추억 점수

https://school.programmers.co.kr/learn/courses/30/lessons/176963

프로그래머스

 

보자마자 Map을 사용해야겠다고 생각이 드는 문제였다.

 

중간에 없는 이름에 대한 예외처리를 생각하지 못하였었다.

 

import java.util.*;
 
class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        int[] answer = new int[photo.length];
 
        //Map으로 풀면 될 것 같은데?
        Map<String, Integer> npList = new HashMap<>();
 
        for(int i = 0; i < name.length; i++) {
            npList.put(name[i], yearning[i]);
        }
 
        int index = 0;
 
        for(String[] names : photo){
            int point = 0;
            //이름이 존재하지 않을 때
            for(int i = 0; i < names.length; i++) {
                if(npList.get(names[i]) == null) {
                    continue;
                } else{
                    point += npList.get(names[i]);
                }
            }
            answer[index] = point;
            index++;
        }
        return answer;
    }
}