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

프로그래머스

2022 KAKAO BLIND RECRUITMENT에서 LV.1으로 출제된 문제이다. 

2024 KAKAO WINTER INTERNSHIP 코테를 봐본 1인으로서 카카오는 HashMap을 활용하는 문제를 좋아하는 듯

 

HaspMap에서 활용되는 메서드들을 정리하기 좋은 문제인 거 같아서 코드블럭별로 자세히 설명을 적어두려고 한다:)

 

👩🏻‍💻 전체 코드는 우선 아래와 같다.

import java.util.*;

class Solution {
    public int[] solution(String[] id_list, String[] report, int k) {
        int[] answer = new int[id_list.length];
        
        HashSet<String> blockUser = new HashSet<>();           	     // 이용 금지 당한 유저
        Map<String, HashSet<String>> reportInfo = new HashMap<>();   // 유저별 신고한 유저 모음
        Map<String, Integer> getReported = new HashMap<>();          // 유저별 신고 당한 수

        for (int i = 0; i < report.length; i++) {
            String[] ppl = report[i].split(" ");

            if (!reportInfo.containsKey(ppl[0]))
                reportInfo.put(ppl[0], new HashSet<>());
            reportInfo.get(ppl[0]).add(ppl[1]);
        }

        for (Map.Entry<String, HashSet<String>> map : reportInfo.entrySet()) {
       	    // 값 제대로 들어갔는지 확인
            // System.out.println(map.getKey() + " : " + map.getValue());
            
            for (String str : map.getValue()) {
                getReported.put(str, getReported.getOrDefault(str, 0) + 1);
                if (getReported.get(str) >= k)
                    blockUser.add(str);
            }
        }

        for (int i = 0; i < id_list.length; i++) {
            String name = id_list[i];
            HashSet<String> values = reportInfo.get(name);

            if (values != null) {
                for (String str : blockUser) {
                    if (values.contains(str))
                        answer[i] += 1;
                }
            }
        }
        return answer;
    }
}

💡 코드블럭 설명

for (int i = 0; i < report.length; i++) {
    String[] ppl = report[i].split(" ");
    
    if (!reportInfo.containsKey(ppl[0]))
        reportInfo.put(ppl[0], new HashSet<>());
    reportInfo.get(ppl[0]).add(ppl[1]);
}

report 배열에 저장된 값을 공백 기준으로 나누어서 HashMap 객체인 reportInfo에 신고한 유저(=ppl[0])를 key값으로, 신고당한 유저(=ppl[1])를 value값으로 추가해준다.

 

이때, 한번도 저장 안 된 유저라면 reportInfo에 Map 객체를 먼저 만들어준 뒤 value에 값을 저장해준다.

for (Map.Entry<String, HashSet<String>> map : reportInfo.entrySet()) {
    // 값 제대로 들어갔는지 확인
    // System.out.println(map.getKey() + " : " + map.getValue());
            
    for (String str : map.getValue()) {
    	getReported.put(str, getReported.getOrDefault(str, 0) + 1);
        if (getReported.get(str) >= k)
        	blockUser.add(str);
    }
}

Map.Entry를 사용하면 Map에 저장된 모든 key-value쌍을 key-value를 갖는 하나의 객체로 얻을 수 있다. -> Map의 각 항목을 구할 수 있다.

 

그래서 Map에 저장된 객체를 한번씩 탐색하면서 해당 객체의 value에 저장된 신고 당한 유저의 신고 당한 횟수를 업데이트 해준다.

Map.Entry 메소드
- getKey()
   Map.Entry 객체의 key값을 반환한다.
- getValue()
   Map.Entry 객체의 value값을 반환한다.

for (int i = 0; i < id_list.length; i++) {
    String name = id_list[i];
    HashSet<String> values = reportInfo.get(name);
    
    if (values != null) {
    	for (String str : blockUser) {
        	if (values.contains(str))
            	answer[i] += 1;
 	}
    }
}

이제 id_list에 저장된 아이디 순서대로 처리 결과 메일을 받은 횟수를 answer 배열에 저장을 해줘야 하기 때문에 id_list에 담긴 순서대로 탐색을 시작한다.

 

본인이 신고한 유저의 아이디를 values에 담고, 누군가를 신고한 기록이 있는 경우라면 본인이 신고한 아이디가 정지된 아이디에 포함되어 있는지 확인을 해준다.

 

만약 포함하는 경우, 처리 결과 메일을 받으므로 answer[i]에 해당하는 값을 1 증가시켜준다.