문제 설명
사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["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]의 원소들은 중복된 값이 들어가지 않습니다.
입출력 예
name | yearning | photo | result |
["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] | [["kai", "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] |
import java.util.HashMap;
class Solution {
public int[] solution(String[] name, int[] yearning, String[][] photo) {
HashMap<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++) {
if (map.containsKey(photo[i][j])) {
answer[i] += map.get(photo[i][j]);
}
}
}
return answer;
}
}
제네릭 타입으로 String과 Integer를 갖는 HashMap을 생성합니다.
for문이 name의 길이만큼 순회합니다.
put 메소드를 이용하여 map에 name의 i 번째 원소와 yearning의 i 번째 원소를 쌍으로 넣습니다.
answer 배열을 생성하고 크기는 photo의 길이로 지정합니다.
바깥 for문이 photo의 길이만큼 순회합니다. 배열의 행을 순회합니다.
안쪽 for문이 photo의 i 번째 원소의 길이만큼 순회합니다. 배열의 열을 순회합니다.
if문에서 containsKey 메소드를 이용하여 map에 photo[i][j] 번째 원소가 있는지 확인합니다. photo 배열에서 앞서 map에 삽입했던, 그리움 점수가 있는 인물이 있는지 확인하는 것입니다.
있다면 answer의 i 번째 원소에 get 메소드를 이용하여 키인 photo[i][j]의 값을 가져와 더하기 할당합니다.
'프로그래머스 > Lv. 1' 카테고리의 다른 글
[프로그래머스 코딩테스트] 명예의 전당 (1)(Java) (0) | 2023.07.24 |
---|---|
[프로그래머스 코딩테스트] 과일 장수(Java) (0) | 2023.07.23 |
[프로그래머스 코딩테스트] 기사단원의 무기(Java) (0) | 2023.07.18 |
[프로그래머스 코딩테스트] 카드 뭉치(Java) (0) | 2023.07.13 |
[프로그래머스 코딩테스트] 소수 만들기(Java) (0) | 2023.07.08 |