신고 결과 받기 - 2022 카카오 신입 문제

 

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

Set과 Map을 사용하여 풀이하였다.

• 유저의 신고받은 횟수를 가지는 reportCountMap
  • Map<String, Integer>
• 유저가 신고한 유저들을 가지는 reportMap
  • Map<String, Set<String>>

중복 신고가 가능하며, 횟수가 한 번만 인정되므로 Set을 사용하였다.

import java.util.*;

class Solution {
    private final String SEPARATE = " ";

    public int[] solution(String[] id_list, String[] report, int k) {
        int[] answer = new int[id_list.length];
        Map<String, Integer> reportCountMap = new HashMap<>();
        Map<String, Set<String>> reportMap = new HashMap<>();
        for (String id : id_list) {
            reportCountMap.put(id, 0);
            reportMap.put(id, new HashSet<>());
        }
        for (String reportInfo : report) {
            String personReporting = reportInfo.split(SEPARATE)[0];
            String personReported = reportInfo.split(SEPARATE)[1];
            if(reportMap.get(personReporting).add(personReported)) {
                reportCountMap.put(personReported, reportCountMap.get(personReported) + 1);
            }
        }

        for(int i = 0 ;i<id_list.length;i++) {
            String id = id_list[i];
            Set<String> reportingList = reportMap.get(id);
            for(String personReported : reportingList) {
                if(reportCountMap.get(personReported) >= k) {
                    answer[i]++;
                }
            }
        }
        return answer;
    }
}