[프로그래머스 Programmers] 신고 결과 받기 - 자바 JAVA

문제 설명

문제 풀기

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

각 유저는 한 번에 한 명의 유저를 신고할 수 있습니다.
신고 횟수에 제한은 없습니다. 서로 다른 유저를 계속해서 신고할 수 있습니다.
한 유저를 여러 번 신고할 수도 있지만, 동일한 유저에 대한 신고 횟수는 1회로 처리됩니다.

k번 이상 신고된 유저는 즉시 게시판 이용이 정지되며, 해당 유저를 신고한 모든 유저에게 정지 사실을 메일로 발송합니다.
게시판 이용이 정지된 유저도 불량 이용자를 신고할 수 있습니다.

생각 정리

이번 문제는 Hash 자료구조를 이용해서 푸는 문제였다.

1. map을 선언하고
   key에 신고된 유저 아이디,
   value에 신고한 유저 아이디를 넣는다.

이 때, value에 들어가는 신고한 유저 아이디는 중복되면 안된다. (문제에서 '동일한 유저에 대한 신고 횟수는 1회로 처리된다' )

2. 신고된 유저 아이디를 순회하면서 정지 기준을 만족하는 유저가 있다면 (신고한 유저아이디가 k명 이상)

   2-1. 신고된 유저에 대한 신고한 유저 아이디를 순회하면서, count를 증가해준다.

이 때, return 하는 배열은 id_list에 담긴 id 순서대로 각 유저가 받은 결과 메일 수를 담아야 한다.

정리된 생각에 대한 논리

(= 내가 문제를 풀어가는데 있어서 어려움을 겪었던 부분)

1. value인 신고한 유저 아이디가 중복되지 않게 하기 위해서, map의 value의 타입을 hash를 이용해서 선언했다.

Map<String, HashSet<String>> map = new HashMap<>();

value에 어떻게 중복없이 넣지 고민을 하다가 검색을 통해 알아낸 방법이다.

2. id_list에 담긴 id 순서대로 메일 수를 담기 위해서, key가 유저 아이디 이고, value가 id_list내에서 해당 유저의 idx를 나타내는 map을 선언했다.

완성된 코드

import java.util.*;
class Solution {
    public int[] solution(String[] id_list, String[] report, int k) {
    
        Map<String, HashSet<String>> map = new HashMap<>();
        Map<String, Integer> idMap = new HashMap<>();
           
         //map 초기화와, id_list에 담긴 id순서대로 유저아이디와 인덱스 담기
        for(int i=0; i<id_list.length; i++) {
            map.put(id_list[i], new HashSet<>());
            idMap.put(id_list[i], i);
        }
        
        //key에는 신고당한 사람, value에는 신고한 사람 넣기 (hashset으로 선언해서 중복 허용x)
        for(String str: report) {
            
            String[] newStr = str.split(" ");
            map.get(newStr[1]).add(newStr[0]);
        }
        
        int[] result = new int[id_list.length];
        
        //key를 순회하며(신고된 사람)
        for(String key: map.keySet()) {
        	//신고횟수가 k 이상일 경우 (해당 key의 value 크기가 k 이상)
            if(map.get(key).size() >= k) {
            	//value를 순회하며, 해당 value(유저 아이디)의 인덱스를 증가 
                for(String singo : map.get(key))
                    result[idMap.get(singo)]++; 
            }
        }
        
        
        return result;
    }
}