[프로그래머스] 추억 점수 - 자바(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]를 반환합니다.

 

 

코드

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

class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        // 결과를 담을 answer 배열
        int[] answer = new int[photo.length];

        // 사진의 추억 점수를 담을 map 선언
        Map<String, Integer> point = new HashMap<>();

        // 사진별 추억점수 map에 담기
        for (int i = 0; i < name.length; i++) {
            point.put(name[i], yearning[i]);
        }

        for (int i = 0; i < photo.length; i++) {
            // 사진에 있는 사람들을 배열에 담기
            String[] person = photo[i];
            int score = 0;

            for (int j = 0; j < person.length; j++) {
                // 배열에 담긴 사진에 찍인 사람의 이름 순차적으로 꺼내기
                String personName = person[j];

                // personName이 map에 key 값으로 하는 경우 해당 사람의 추억점수 더하기
                if (point.containsKey(personName)) {
                    score += point.get(personName);
                }

            }
            answer[i] = score;
        }

        return answer;
    }
}

코드 설명

주석 참고

• photo 배열에는 사진에 찍힌 사람들의 이름이 담겨 있으며, 추억 점수를 계산할 사람들과 해당 사람들의 점수를 표현한 배열 name, yearning이 존재한다.
• name 배열에 존재하는 사람들을 yearning 배열을 통해 계산할 점수를 얻고 photo에 존재하는 사람의 이름이 name에 존재할 때마다 yearning에 해당하는 점수를 증가시키는 문제다.
• containsKey() 메서드를 통해 photo 배열을 순회하면서 각 이름이 point의 키로 존재하는지 확인할 수 있다.
• photo 배열에 존재하는 이름이 map의 key로 존재하는 경우 해당 value를 꺼내 score 값에 더하여 answer 배열에 담는다.