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

---

입출력 예

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

설명 생략

---

풀이

import java.util.*; // HashMap 등을 사용하기 위해 import

class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        
        // 1️⃣ 이름별 그리움 점수를 빠르게 찾기 위한 해시맵 생성
        // 예: { "may"=5, "kein"=10, "kain"=1, "radi"=3 }
        Map<String, Integer> scoreMap = new HashMap<>();
        
        // 2️⃣ name과 yearning 배열의 같은 인덱스를 이용해 맵에 저장
        for (int i = 0; i < name.length; i++) {
            scoreMap.put(name[i], yearning[i]);
        }
        
        // 3️⃣ 결과값(각 사진의 추억 점수)을 담을 int 배열 생성
        int[] answer = new int[photo.length];
        
        // 4️⃣ 각 사진에 대해 반복 (사진 하나씩 확인)
        for (int i = 0; i < photo.length; i++) {
            
            int sum = 0; // 현재 사진의 추억 점수를 저장할 변수
            
            // 5️⃣ 사진 속 인물들을 하나씩 확인
            for (String person : photo[i]) {
                
                // 6️⃣ 해당 인물이 그리움 점수 목록에 있으면 (맵에 존재하면)
                if (scoreMap.containsKey(person)) {
                    // 7️⃣ 그 인물의 점수를 합산
                    sum += scoreMap.get(person);
                }
                // 없으면 자동으로 0점이므로 아무것도 안 함
            }
            
            // 8️⃣ 계산된 사진의 총 추억 점수를 결과 배열에 저장
            answer[i] = sum;
        }
        
        // 9️⃣ 모든 사진에 대한 계산이 끝났으므로 결과 배열 반환
        return answer;
    }
}

import java.util.*;
import java.util.stream.*;

class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        
        // 1️⃣ 이름(name)과 점수(yearning)를 하나의 Map으로 변환
        // IntStream.range(0, name.length) → 0부터 name.length-1까지 숫자 스트림 생성
        // mapToObj(i -> Map.entry(name[i], yearning[i])) → (이름, 점수) 쌍으로 바꿈
        // collect(Collectors.toMap(...)) → Map<String, Integer>으로 수집
        Map<String, Integer> map = IntStream.range(0, name.length)
            .mapToObj(i -> Map.entry(name[i], yearning[i]))
            .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
        
        
        // 2️⃣ 각 사진(photo 배열)에 대해 추억 점수를 계산
        // Arrays.stream(photo) → 사진 하나씩 스트림으로 변환
        // mapToInt(strings -> ...) → 각 사진의 인물 목록(strings)에 대해 합산 점수를 구함
        // 내부에서 Arrays.stream(strings)으로 각 인물 이름을 돌며 map에서 점수 조회
        // map.getOrDefault(value, 0) → 이름이 없으면 0점 처리
        // sum() → 한 사진의 점수 합
        // 마지막에 toArray()로 결과를 int[]로 변환
        return Arrays.stream(photo)
            .mapToInt(strings -> 
                Arrays.stream(strings)
                      .mapToInt(person -> map.getOrDefault(person, 0))
                      .sum()
            )
            .toArray();
    }
}