[프로그래머스] 신고 결과 받기 (JAVA 풀이)

 

문제 접근

 

https://school.programmers.co.kr/learn/courses/30/lessons/92334

프로그래머스

 

해시를 사용하면 쉽게 해결할 수 있는 문제였다!

 

여러 방법이 있겠지만 가장 직관적으로 각 유저별로 누가 신고를 했는지 List의 정보를 담고있는 Map 하나와

각 유저별로 처리 결과 메일을 몇 개를 받아야하는지를 기록하는 또 다른 Map 하나를 이용하여 문제를 풀이하는 방향성을 생각했다.

 

문제 풀이

 

한 유저를 여러 번 신고할 수도 있지만, 동일한 유저에 대한 신고 횟수는 1회로 처리되기 때문에 Set을 이용해서 이런 중복 신고를 제거해줬다.

 

=> 그리고 위에서 말한대로 Map 2개를 선언하고 각 유저별로 초기화해줬음

 

이렇게 반복문을 모두 돌게되면 reportMap에 각 유저별로 가지고있는 List의 size가 신고를 당한 횟수임

 

=> 이 size가 k 이상이라면 해당 유저를 신고한 모든 이용자(size가 k가 넘는 List에 담겨있는 유저들)의 resultMap에 +1을 해주며 처리 결과 메일 개수를 +1 해줌

 

이제 이 resultMap에는 각 유저별로 처리 결과 메일을 몇개 받아야하는지가 들어있으므로 이를 그대로 정답 배열에 담아서 반환해주면 끝남

 

전체 코드

 

import java.util.*;

class Solution {
    public int[] solution(String[] id_list, String[] report, int k) {
        
        Set<String> set = new HashSet<>(); // 중복 신고를 없애기 위한 Set
        Map<String, List<String>> reportMap = new HashMap<>(); // 각 유저별 신고 당한 목록
        Map<String, Integer> resultMap = new HashMap<>(); // 처리 결과 메일을 받을 수
        
        // Map 초기화
        for(int i = 0; i < id_list.length; i++) {
            reportMap.put(id_list[i], new ArrayList<>());
            resultMap.put(id_list[i], 0);
        }
        
        // report의 중복 신고 제거
        for(int i = 0; i < report.length; i++) {
            set.add(report[i]);
        }
        
        // 각 유저별로 해당 유저를 신고한 유저의 List를 갱신
        for(String s : set) {
            
            String[] temp = s.split(" ");
            reportMap.get(temp[1]).add(temp[0]);
        }
        
        // 신고 횟수 k 넘은 유저를 신고한 모든 이용자에게 처리 결과 메일 개수를 +1 해줌
        for(Map.Entry<String, List<String>> entrySet : reportMap.entrySet()) {
            
            List<String> list = entrySet.getValue();
            
            if(list.size() >= k) {
                for(int i = 0; i < list.size(); i++) {
                    resultMap.put(list.get(i), resultMap.get(list.get(i)) + 1);
                }
            }
        }
        
        int[] answer = new int[id_list.length];
        
        // 각 유저별 처리 결과를 answer 배열에 저장
        for(int i = 0; i < id_list.length; i++) {
            answer[i] = resultMap.get(id_list[i]);
        }
        
        return answer;
    }
}