[연습문제] 추억 점수

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]의 원소들은 중복된 값이 들어가지 않습니다.

---

입출력 예

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]	[["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]

---

입출력 예 설명
입출력 예 #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

• 설명 생략

---

사진마다 찍혀 있는 인원들의 추억 점수를 더해 총 추억 점수가 몇 점인지 구하는 문제입니다.

 

저는 해시 맵과 딕셔너리를 사용해서 인물마다의 점수를 기록하고, 반복문으로 그 값을 가져와서 더하는 식으로 문제를 해결했습니다.

다른 사람들의 풀이를 좀 보니 인덱스를 사용해서도 문제를 풀 수 있었는데, 제 기준으로는 한 번 보고 바로 이해하기 쉽지 않아서 단계적으로 풀었습니다.

 

정답 코드를 작성해보겠습니다.

 

1. 자바

import java.util.HashMap;

class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        // 사진의 개수를 기록합니다.
        int n = photo.length, idx = 0;
        int[] answer = new int[n];
        // 해시맵으로 인물 - 추억 점수를 저장합니다.
        HashMap<String, Integer> scores = new HashMap<String, Integer>();
        for (int i = 0; i < name.length; i++) {
            scores.put(name[i], yearning[i]);
        }
        // 사진별로 인물을 확인합니다.
        for (String[] people: photo) {
            // 사진별 점수를 기록하는 score
            int score = 0;
            for (String p: people) {
                score += scores.getOrDefault(p, 0); // 기본값 0을 설정
            }
            // 배열에 하나씩 저장합니다.
            answer[idx++] = score;
        }
        return answer;
    }
}

 

2. 파이썬

from collections import defaultdict
def solution(name, yearning, photo):
    # 정답을 기록할 리스트
    answer = []
    # 그리움 점수를 기록할 딕셔너리
    scores = defaultdict(int)
    # 인물별 그리움 점수를 기록합니다.
    for i, person in enumerate(name):
        scores[person] = yearning[i]
    # 반복문을 통해 각 사진별 점수를 합산해 정답에 기록합니다.
    for picture in photo:
        score = sum(scores[person] for person in picture)
        answer.append(score)
    # 정답을 반환합니다.
    return answer

 

 

깔끔한 코드, 짧은 코드도 좋아하지만, 역시 가독성이 가장 중요한 것 같습니다.