[프로그래머스] 추억점수
문제
사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다.
사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다.
예를 들어 사진 속 인물의 이름이 ["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]의 원소들은 중복된 값이 들어가지 않습니다.
입출력 예
| ["may", "kein", "kain", "radi"] | [5, 10, 1, 3] | [["may", "kein", "kain", "radi"],["may", "kein", "brin", "deny"], ["kon", "kain", "may", "coni"]] | [19, 15, 6] |
|---|---|---|---|
| ["kali", "mari", "don"] | [11, 1, 55] | [["kali", "mari", "don"], ["pony", "tom", "teddy"], ["con", "mona", "don"]] | [67, 0, 55] |
| ["may", "kein", "kain", "radi"] | [5, 10, 1, 3] | [["may"],["kein", "deny", "may"], ["kon", "coni"]] | [5, 15, 0] |
풀이
문제의 흐름을 그대로 따라가면 되었다.
이름과 점수를 매칭할 수 있도록 해야 했다.
나중에 for로 한개씩 접근해서 찾아오면 시간이 많이 걸리기 때문이다.
이름의 인덱스와 점수의 인덱스가 서로 공유가 되므로 Map에 이름과 인덱스를 담아두었다.
인덱스 대신 점수를 저장해도 무리없이 동일하게 풀이가 가능하다.
이후 photo배열에 요소들을 하나씩 가져와서 더해주어 계산만 하면 되었다.
점수가 없는 이름일 경우 null을 통해 분기처리를 해야 됨을 잊지 말자.
import java.util.*;
import java.lang.*;
class Solution {
public int[] solution(String[] name, int[] yearning, String[][] photo) {
HashMap<String, Integer> map = new HashMap<>();
int resultSize = photo.length;
int[] answer = new int[resultSize];
for(int i = 0; i < name.length; i++) {
map.put(name[i], i);
}
for(int i = 0; i < resultSize; i++) {
int sum = 0;
for(int j = 0; j < photo[i].length; j++) {
if (map.get(photo[i][j]) == null) {
continue;
}
sum += yearning[map.get(photo[i][j])];
}
answer[i] = sum;
}
return answer;
}
}