사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이
["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
의 원소의 길이 ≤ 7name
의 원소들은 알파벳 소문자로만 이루어져 있습니다.name
에는 중복된 값이 들어가지 않습니다.- 1 ≤
yearning[i]
≤ 100yearning[i]
는 i번째 사람의 그리움 점수입니다.
- 3 ≤
photo
의 길이 ≤ 100
- 1 ≤
photo[i]
의 길이 ≤ 100- 3 ≤
photo[i]
의 원소(문자열)의 길이 ≤ 7photo[i]
의 원소들은 알파벳 소문자로만 이루어져 있습니다.photo[i]
의 원소들은 중복된 값이 들어가지 않습니다.
- 입출력 예 #1 : 첫 번째 사진 속 "may", "kein", "kain", "radi"의 그리움 점수를 합치면 19(5 + 10 + 1 + 3)점 입니다. 두 번째 사진 속 그리워하는 사람들인 "may"와 "kein"의 그리움 점수를 합치면 15(5 + 10)점입니다. 세 번째 사진의 경우 "kain"과 "may"만 그리워하므로 둘의 그리움 점수를 합한 6(1 + 5)점이 사진의 추억 점수입니다. 따라서 [19, 15, 6]을 반환합니다.
- 입출력 예 #2 : 첫 번째 사진 속 그리워하는 사람들인 "kali", "mari", "don"의 그리움 점수를 합치면 67(11 + 1 + 55)점입니다. 두 번째 사진 속엔 그리워하는 인물이 없으므로 0점입니다. 세 번째 사진 속 그리워하는 사람은 "don"만 있으므로 55점입니다. 따라서 [67, 0, 55]를 반환합니다.
- 입출력 예 #3 : 설명 생략
✅ 첫번째 for문 : 이차원 배열
photo
를 각 사진으로, 즉 행 단위로 구분한다.
두번째 for문 : 각 사진으로 구분한photo[i]
를 사진 속 인물으로, 즉 열 단위로 구분한다.
세번째 for문 : 한 사진의 인물photo[i][j]
가name[]
속 요소에 있다면 그에 해당하는 인물의 그리움 점수yearning[k]
를 각 사진의 그리움 점수answer[i]
에 누적합시킨다.
class Solution {
public int[] solution(String[] name, int[] yearning, String[][] photo) {
int[] answer = new int[photo.length];
for(int i=0;i<photo.length;i++) {
for(int j=0;j<photo[i].length;j++) {
for(int k=0;k<name.length;k++) {
if(photo[i][j].equals(name[k])) {
answer[i] += yearning[k];
}
}
}
}
return answer;
}
}
➕ 다른 사람의 코드 : 내 풀이는 삼중 for문을 사용하다보니 속도 측면에서 좀 찝찝했었는데, HashMap을 사용하여
name[]
과yearning[]
을 한 번에 저장해서 for문 하나를 줄인 코드가 있어서 실행해보니 속도가 훨씬 빨라졌다!
이름은 중복값이 없고, 그리움 점수에는 중복값이 있을 수 있으므로<이름, 그리움 점수>
형태의 HashMap을 생성하여 이름과 그에 대한 그리움 점수를 저장하였다.
이후 이중 for문은 위의 첫번째, 두번째 코드와 같다! 단, 두번째 for문에서 다시 for문을 사용하지 않고,getOrDefault(photo[i][j], 0)
메서드를 사용하여photo[i][j]
가map
에 존재하는 이름이면 해당 인물의 그리움 점수를, 존재하지 않으면 0을answer[i]
에 누적합시킨다.
import java.util.*;
class Solution {
public int[] solution(String[] name, int[] yearning, String[][] photo) {
Map<String, Integer> map = new HashMap<>();
for(int i=0;i<name.length;i++) {
map.put(name[i], yearning[i]);
}
int[] answer = new int[photo.length];
for(int i=0;i<photo.length;i++) {
for(int j=0;j<photo[i].length;j++) {
answer[i] += map.getOrDefault(photo[i][j], 0);
}
}
return answer;
}
}