프로그래머스 신고결과받기

문제 설명

신입사원 무지는 게시판 불량 이용자를 신고하고 처리 결과를 메일로 발송하는 시스템을 개발하려 합니다. 무지가 개발하려는 시스템은 다음과 같습니다.

각 유저는 한 번에 한 명의 유저를 신고할 수 있습니다.
신고 횟수에 제한은 없습니다. 서로 다른 유저를 계속해서 신고할 수 있습니다.
한 유저를 여러 번 신고할 수도 있지만, 동일한 유저에 대한 신고 횟수는 1회로 처리됩니다.
k번 이상 신고된 유저는 게시판 이용이 정지되며, 해당 유저를 신고한 모든 유저에게 정지 사실을 메일로 발송합니다.
유저가 신고한 모든 내용을 취합하여 마지막에 한꺼번에 게시판 이용 정지를 시키면서 정지 메일을 발송합니다.
다음은 전체 유저 목록이 ["muzi", "frodo", "apeach", "neo"]이고, k = 2(즉, 2번 이상 신고당하면 이용 정지)인 경우의 예시입니다.

각 유저별로 신고당한 횟수는 다음과 같습니다.

위 예시에서는 2번 이상 신고당한 "frodo"와 "neo"의 게시판 이용이 정지됩니다. 이때, 각 유저별로 신고한 아이디와 정지된 아이디를 정리하면 다음과 같습니다.

문제풀이

1. 사용자id, 신고당한 id HashMap 생성
2. 신고 report를 HashSet으로 받아 중복제거
3. for 돌며 HashSet 공백 split으로 신고횟수 map에 추가
4. for 돌며 HashSet 사용자 id에 신고한 id 정지 받은 횟수 추가

코드

import java.util.*;
class Solution {
    public int[] solution(String[] id_list, String[] report, int k) {
        int[] answer = new int[id_list.length];
        HashMap<String,Integer> id = new HashMap<>();
        HashMap<String,Integer> warn = new HashMap<>();
        HashSet<String> Set = new HashSet<>();
        for(String x : id_list){
            id.put(x, 0);
            warn.put(x,0);
        }
        for(String x : report){
            Set.add(x);
        }
        for(String x : Set){
            String[] arr = x.split(" ");
            warn.put(arr[1],warn.get(arr[1])+1);
        }
         for(String x : Set){
            String[] arr = x.split(" ");
            if(warn.get(arr[1])>=k)id.put(arr[0],id.get(arr[0])+1);
        }
        for(int i=0;i<answer.length;i++){
            System.out.println(id_list[i]+" "+warn.get(id_list[i]));
            answer[i]=id.get(id_list[i]);
        }
        return answer;
    }
}