[프로그래머스] 추억 점수 자바 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]의 원소들은 중복된 값이 들어가지 않습니다.

---

 

이 문제는 그리워하는 사람과 점수를 가지고 사진 속 인물들 중 그리워하는 사람들의 그리움 점수 총합을 구해 배열로 반환하라는 문제이다. 

 

우선은 그리운 사람과 점수를 <이름-그리움점수>의 map으로 만들어 주었다.

그리고 photo배열을 돌면서 해당 사진의 그리운 사람 점수를 더하도록 했고, 그 값은 containsKey를 사용하여 그리운 사람일 경우에만 그 점수를 더할 수 있도록 하였다. 

 

 

전체코드

import java.util.*;
class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        
        int[] answer = new int[photo.length];
        Map<String,Integer> map = new HashMap<>();        
        for(int i=0;i<name.length;i++){
            map.put(name[i],yearning[i]);
        }
        
        
        int idx=0;
        for(String[] pho : photo){
            for(String man : pho){
                if(map.containsKey(man))
                    answer[idx]+=map.get(man);
            }
            idx++;
        }
           
        return answer;
    }
}