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

링크 : https://school.programmers.co.kr/learn/courses/30/lessons/92334

프로그래머스

---

문제 풀이

import java.util.*;

class Solution {
    public int[] solution(String[] id_list, String[] report, int k) {
        
        int[] answer = new int[id_list.length];
        Map<String, HashSet<String>> reportedMap = new HashMap<>(); // 중복을 제거하기 위해 HashSet 사용
        Map<String, Integer> idxMap = new HashMap<>();
        
        for(int i=0; i<id_list.length; i++) {
            reportedMap.put(id_list[i], new HashSet<>());
            idxMap.put(id_list[i], i);
        }
        
        for(int i=0; i<report.length; i++) {
            String[] str = report[i].split(" "); // 띄어쓰기 기준 문자열 분리
            String a = str[0]; // 신고한 ID
            String b = str[1]; // 신고된 ID
            
            reportedMap.get(b).add(a);
        }
        
        for(int i=0; i<id_list.length; i++) {
            HashSet<String> send = reportedMap.get(id_list[i]);
            if(send.size() >= k) {
                for(String name : send) {
                    answer[idxMap.get(name)]++;
                }
            }
        }
        return answer;
    }
}