[프로그래머스] lv1 추억 점수 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 함수를 완성해 주세요.

해설

해당 문제는 name에 해당하는 추억점수를 HashMap에 넣어주고 photo를 반복문을 사용하여 순차적으로 탐색하여 나온 name을 HashMap에서 꺼내 합산해 주고 배열에 차곡차곡 넣어주면 된다. 또한, HashMap에 없는 name은 0 처리해 주면 간단하게 해결된다.

문제 풀이

문제 풀이는 보자마자 해시로 풀면 될 것 같단 생각이 들어 브루트포스 방식으로 먼저 접근하지 않고 바로 해시로 풀었다.

참고로 브루트포스로 풀면 시간복잡도는 O(N^3)이 나오며, 해시 방식은 O(N^2)으로 풀어낼 수 있다.

1. 해시 방식

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++) {
            String nameItem = name[i];
            int point = yearning[i];
            
            map.put(nameItem, point);
        }
        
        for (int i = 0; i < photo.length; i++) {
            for (String item: photo[i]) {
                answer[i] += map.getOrDefault(item, 0);
            }
        }
        
        return answer;
    }
}