자바 - 프로그래머스 / 추억 점수

https://school.programmers.co.kr/learn/courses/30/lessons/176963

프로그래머스

LV 1

 

구현 방법

제한 사항

3 ≤ name의 길이 = yearning의 길이≤ 100

3 ≤ photo의 길이 ≤ 100

1 ≤ photo[i]의 길이 ≤ 100

 

이므로 photo를 이중 포문 돌려도 된다. 또한 name과 yearning을 해시 맵으로 만들면 이름 비교는 O(1)으로 해결 됨.

따라서 HashMap<String, Integer> 필요하다. 문제는 해시맵에서 이 key값이 존재 하는지 확인하는 과정이 필요한데,

이건 HashMap의 containsKey() 메서드를 통해 확인할수 있다.

 

코드

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

class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        HashMap<String, Integer> hm = new HashMap<>();
        
        for (int i = 0; i < name.length; i++) {
            hm.put(name[i], yearning[i]);
        }
        
        int[] answer = new int[photo.length];
        
        Arrays.fill(answer, 0);
    
        for (int i = 0; i < photo.length; i++) {
            for(String cur : photo[i]) {
                if (hm.containsKey(cur)) {
                    answer[i] += hm.get(cur);
                }
            }
        }
        
        return answer;
    }
}

 

느낀 점

이것도 레벨 1 이라 문제 자체는 쉽고, 자바 기본 라이브러리(HashMap의 contiansKey) 익히는데 도움이 되어서 포스팅 했다.

 

참고 블로그

https://hianna.tistory.com/574

[Java] HashMap에 특정 key 존재 여부 확인하기