[프로그래머스/JAVA] 신고 결과 받기

👇 문제 URL 👇

프로그래머스

 

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;

public static int[] solution(String[] id_list, String[] report, int k) {
    int[] answer = new int[id_list.length];

    // HashMap : Key-value, hashSet: set 중 성능이 가장 좋음
    // key : 신고된 ID, value : 신고한 ID set
    Map<String, HashSet<String>> reportMap = new HashMap<>();

    // key : 신고한 ID, value : 메일 받는 횟수
    Map<String, Integer> resultMap = new HashMap<>();

    for (int i = 0; i < id_list.length; i++) {
        String name = id_list[i];
        reportMap.put(name, new HashSet<>());
        resultMap.put(name, 0);
    }

    for(String user_report : report){
        String[] reportArr = user_report.split(" ");
        reportMap.get(reportArr[1]).add(reportArr[0]); //신고한 ID를 HashSet에 추가해줌
    }

    //reportMap 데이터 출력
    /*reportMap.forEach((key, value) -> {
        System.out.print(key +":" );
        value.forEach((str)->System.out.print(str +"," ));
        System.out.println();
    } );*/

    //신고한 ID별 메일 받는 횟수 추가
    reportMap.forEach((key,value) -> {

        //신고당한 횟수가 k회 이상이라면 메일 발송
        if(value.size() >= k){

            value.forEach((str) -> {

                //신고자명 찾아서 메일 횟수 추가
                int cnt = resultMap.get(str).intValue() +1;
                resultMap.replace(str, cnt);

            });
        }
    });

    //resultMap 데이터 출력
    /*resultMap.forEach((key, value) -> {
        System.out.println(key +":" + value);
    } );*/


    for(int i=0; i< id_list.length; i++){
        answer[i] = resultMap.get(id_list[i]).intValue();
    }

    return answer;
}

테스트도 다 통과했다!