프로그래머스) 추억 점수(level 1) - Java(자바)

문제 설명

사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["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 함수를 완성해주세요.

---

입출력 예

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]

---

문제 풀이

import java.util.Map;
import java.util.HashMap;
import java.util.Arrays;

class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        Map<String, Integer> scoreMap = new HashMap<>();
        
        for(int i = 0; i < name.length; i++){
            scoreMap.put(name[i], yearning[i]);
        }
        
        int[] answer = new int[photo.length];    // 사진 수만큼 결과 배열 생성
        
        for(int i = 0; i < photo.length; i++){   // 각 사진에 대해 반복
            int sum = 0;
            for(String person : photo[i]){       // 사진 속 인물들을 순회
                sum += scoreMap.getOrDefault(person, 0);  // 이름이 있으면 점수 더하고, 없으면 0
            }
            answer[i] = sum;   // 총 점수 저장
        }
        return answer;
    }
}

class Main{
    public static void main(String[] args){
        Solution s = new Solution();    // solution 객체 생성
        String[] name = {"may", "kein", "don", "tony"};
        int[] yearning = {5, 10, 1, 3};    
        String[][] photo = {
            {"may", "kein", "kain"},
            {"kali", "mari", "don", "tony"}
        };
        
        int[] result = s.solution(name, yearning, photo);   // 호출 후 결과 저장
    }
}

scoreMap : 사람 이름 / 그리움 점수를 저장하는 맵

    - ex) { "may"=5, "kein"=10, "don"=1, "tony"=3 }

answer : 사진별 추억 점수 저장하는 배열

scoreMap.getOrDefault(person, 0) :

    - "kain"이나 "kali"처럼 name 배열에 없는 인물은 0점으로 간주

 

결과 : 

첫 번째 사진 : 5(may) + 10(kein) + 0(kain) = 15

두 번째 사진 : 0(kali) + 0(mari) + 1(don) + 3(tony) = 4

---

 

프로그래머스