[프로그래머스] 신고 결과 받기 (Java)

 

프로그래머스

이 문제가 왜 레벨1인지는 사실 아직도 잘 이해가 안 된다. 레벨1.5 ~ 2는 되어야하지 않을까?

 

자바에서의 HashSet과 HashMap의 활용에 익숙해질때까지 며칠동안 다른 문제들을 풀면서 이 문제도 간간히 눈으로 봤는데, 어느 순간 한 번에 로직을 구성할 수 있을 것 같아 기존 코드를 지우고 새로 작성했다.

 

- 문제 해결 과정에서 3중 for문으로 배열을 탐색해야하여 몇몇 테스트 케이스에서 시간초과가 발생함

- 신고된 리스트와 횟수를 배열을 활용하여 저장하고 탐색하기보다는 Map과 Set을 활용하여 탐색하는 것이 훨씬 효율적임

- 특히 신고리스트 HashMap의 value 타입을 HashSet으로 선언하여 사용할 경우, 신고 횟수와 메일 발송 리스트 둘 다를 한 번에 산출할 수 있어 배열을 사용할 때보다 훨씬 편했음

 

Note

- 익숙해진 메서드 : Set.contains(), String.split()

- HashMap의 value에 HashSet<String>을 선언할 수 있음을 알게되었다.

- HashSet은 for문 탐색도 가능하다는 걸 알게되었다. 다만, 가급적 key로 접근하는 것이 효율적인 듯하다.

import java.util.*;

class Solution {
    public int[] solution(String[] id_list, String[] report, int k) {
       
        int[] answer = new int[id_list.length];
        
        HashMap<String, HashSet<String>> reported = new HashMap<>();
        
        for(String id : id_list){
            reported.put(id, new HashSet<String>());
        }
        
        for(String rep : report){
            reported.get(rep.split(" ")[1]).add(rep.split(" ")[0]);
        }
        
        for(String id : id_list){
            
            HashSet<String> reporters = reported.get(id);
            
            if(reporters.size() >= k){
                for(int i = 0; i < id_list.length; i++){
                    if(reporters.contains(id_list[i])){
                        answer[i]++;
                    }
                }
            }  
        }
        
        return answer;
    }
}