프로그래머스 - 추억 점수

홍성진·2023년 3월 31일
0

프로그래머스 - 추억 점수

인물들의 sequence가 존재할 때, name에는 i번째 인물의 이름이, yearning에는 i번째 인물에 대한 추억 점수가 들어 있습니다. 그리고 photo의 원소인 각 사진에는 임의의 인물 이름들이 들어 있습니다. 이 때, 사진별 추억 점수합을 return하는 문제입니다. 동명이인이 없다는 제약조건이 있었기 때문에 간편하게 dictionary를 이용해 해결할 수 있었습니다.

import java.util.HashMap;

class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        int[] answer = new int[photo.length];
        
        HashMap<String, Integer> hashMap = new HashMap<>();
        for (int i = 0; i < name.length; i++) {
            hashMap.put(name[i], yearning[i]);
        }
        
        for (int i = 0; i < photo.length; i++) {
            for (String n : photo[i]) {
                if (hashMap.get(n) != null) {
                    answer[i] += hashMap.get(n);
                }
            }
        }
        
        return answer;
    }
}

0개의 댓글