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

프로그래머스 문제 링크 : https://programmers.co.kr/learn/courses/30/lessons/92334

코딩테스트 연습 - 신고 결과 받기

 

문제

각 유저는 한 번에 한명의 유저를 신고할 수 있다.

신고 횟수에 제한은 없다. 서로 다른 유저를 계속해서 신고할 수 있다.

한 유저를 여러 번 신고할 수도 있지만, 동일한 유저에 대한 신고 횟수는 1회로 간주한다.

k번 이상 신고된 유저는 게시판 이용이 정지되며, 해당 유저를 신고한 모든 유저에게 정지 사실을 메일로 발송한다.

유저가 신고한 모든 내용을 취합하여 마지막에 한꺼번에 게시판 이용 정지를 시키면서 정지 메일을 발송한다.

접근

k번 이상 신고된 유저의 리스트를 확보하고, 해당 유저를 신고한 유저 리스트를 만든다. 신고한 유저들에게 정지 메일을 발송하게 된다.

1. 신고당한 사람(string)과 신고당한 횟수(int)를 카운팅하는 key-value HashMap을 만든다.

2. 신고한 사람(string)과 신고 대상들(set)로 구성된 key-value HashMap을 만든다.

 

import java.util.*;

class Solution {
    public static int[] solution(String[] id_list, String[] report, int k) {

        int[] answer = new int[id_list.length];

        HashMap<String, Integer> countMap = new HashMap<>(); // 신고당한사람-신고당한횟수
        HashMap<String, Integer> resultMap = new HashMap<>(); //

        HashMap<String, Set<String>> reportSet = new HashMap<>(); // 신고한사람 - 신고대상들


        for(String str : report) {
            String reporter = str.split(" ")[0]; // 신고한사람
            String reportee = str.split(" ")[1]; // 신고당한사람

            Set<String> set = reportSet.getOrDefault(reporter, new HashSet<>());
            if(!set.contains(reportee)) {
                // 신고당한사람-신고당한횟수
                countMap.put(reportee, countMap.getOrDefault(reportee, 0)+1);
                set.add(reportee);
                reportSet.put(reporter, set);
            }

        }


        for(int i=0; i<id_list.length; i++) {
            String name = id_list[i];
            // 신고한사람이 있으면, 신고대상들 리스트로 가져옴
            if(reportSet.get(name)!=null) {
                for(String s : reportSet.get(name)) {
                    // 신고대상들 중 k번 이상인 사람들
                    if(countMap.get(s) >= k) {
                        //resultMap.put(name, resultMap.getOrDefault(name, 0)+1);
                        answer[i] = answer[i]+1;
                    }
                }
            } else {
                //resultMap.put(name, 0);
            }
        }

        return answer;
    }
}