인물들의 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;
}
}