[연습문제][lv.1] 추억 점수 - 자바(Java)
문제 링크
https://school.programmers.co.kr/learn/courses/30/lessons/176963
문제
사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["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]의 원소들은 중복된 값이 들어가지 않습니다.
✅ 풀이 과정
0. 자료구조 / 알고리즘 선택
간단한 구현문제이므로 그대로 구현하면 된다.
사람의 이름은 중복이 없기 때문에 빠른 조회를 위해 HashMap을 사용한다.
1. 복잡도 고려
사진의 개수만큼 반복하고
> 사진마다 사람의 개수만큼 반복하므로
최악의 경우 100 * 100 = 10000밖에 되지 않는다.
2. 수도코드/ 알고리즘 생각
> 문제 그대로 구현해주면 됨
✅ 내가 쓴 정답코드
import java.util.*;
class Solution {
public int[] solution(String[] name, int[] yearning, String[][] photo) {
//name은 중복되지 않으므로, key = name, value = yearning을 넣은 HashMap을 생성한다.
//0) 자료 초기화
Map<String, Integer> personYearningMap = new HashMap<String, Integer>();
int personNum = name.length;
for (int i = 0; i < personNum; i++) {
personYearningMap.put(name[i], yearning[i]);
}
int photoNum = photo.length;
int[] result = new int[photoNum];
//1) photo의 개수만큼 반복한다 -> 100
for (int i = 0; i < photoNum ; i++) {
int photoYearningSum = 0;
//각 사진에 존재하는 사람만큼 반복한다. -> 100
for (String person : photo[i]) {
Integer score = personYearningMap.get(person);
if (score != null)
photoYearningSum += score;
}
result[i] = photoYearningSum;
}
return result;
}
}
references
프로그래머스