[프로그래머스/해시] Lv1. 신고 결과 받기(자바)

 

해시 

import java.util.*;

// 1. 신고 내역을 set 으로 만든다.
// 2. 신고 결과 내역을 map 으로 만든다. (신고 대상 아이디: 신고한 사람 아이디 목록)
// 3. 정지된 계정을 찾는다.
// 4. 해당 계정을 신고한 아이디 목록을 map 으로 정리 (신고한 사람 아이디: count)

class Solution {
    
    public int[] solution(String[] id_list, String[] report, int k) {
        int[] answer = new int[id_list.length];
        // 1. 중복 제거
        HashSet<String> reportSet = new HashSet<String>();
        for(String rep : report){
            reportSet.add(rep);
        }

        // 2. notifyListHash 만들기
        HashMap<String, ArrayList<String>> notifyListHash = new HashMap();
        for(String rep : reportSet) {
            String[] data = rep.split(" ");
            String reporter = data[0];
            String reportee = data[1];

            ArrayList<String> reporterList = notifyListHash.getOrDefault(reportee, new ArrayList());
            reporterList.add(reporter);
            notifyListHash.put(reportee, reporterList);
        }
        // 3. notifyListHash 를 기반으로 reporterHash 만들기
        HashMap<String,Integer> reporterHash = new HashMap();
        for(ArrayList<String> notifyList : notifyListHash.values()) {
            if(notifyList.size() >= k) {
                for(String reporter : notifyList) {
                    reporterHash.put(reporter, reporterHash.getOrDefault(reporter, 0) + 1);
                }
            }
        }

        // 4. reporterHash 를 기반으로 answer 채운다
        for(int i=0; i <id_list.length; i++) {
            answer[i] = reporterHash.getOrDefault(id_list[i], 0);
        }

        return answer;
    }
    
}

참고 : 개발자로 취직하기