[프로그래머스] [Level1] 신고결과받기 JAVA

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

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

[문제]

[풀이]

신고기준 횟수k 를 넘어 신고를 당한사람을 블랙처리하고 그 결과를 신고한 사람에게 배열형태로 알려주는 문제이다.

우선 Map를 사용함으로써 <신고당한사람, 신고한사람들> 의 형태로 상황을 유지시킨다. 
같은사람이 여러번 신고한 경우를 필터링하기위해 key를 신고당한사람, value를 HashSet으로 함으로써 중복을 기본적으로 걸러낸다.

HashMap은 기본적으로 데이터가 들어간 순서대로 entry가 돌지 않기때문에 신고한 사람 순서대로 결과를 반환하기위해서 LinkedHashMap을 사용하여 반환값을 설정해주는 response 를 따로 만든다. 

import java.util.*;
import java.io.*;
import java.util.stream.Collector;
import java.util.stream.Collectors;
 
public class Main {
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] id_list = {"con", "ryan"};
        String[] report = {"ryan con", "ryan con", "ryan con", "ryan con"};
        int k = 3;
 
        Solution solution = new Solution();
        int[] answer = solution.solution(id_list, report, k);
        System.out.println(Arrays.toString(answer));
    }
}
 
class Solution {
    public int[] solution(String[] id_list, String[] report, int k) {
        int[] answer = {};
        Board board = new Board(id_list, k);
        board.update(report);
        return board.getResponse();
    }
 
    static class Board {
        private int criteria;
        // key: userName, value:subjects
        private HashMap<String,HashSet<String>> state = new HashMap<>();
 
        //key: subject, value:response
        private Map<String,Integer> response = new LinkedHashMap<>();
 
        public Board(String[] id_list,int criteria) {
            this.criteria = criteria;
 
            for (String id : id_list) {
                // 같은사람이 여러번 신고한 것을 제거하기위해 HashSet 을 사용
                state.put(id,new HashSet<>());
                response.put(id,0);
            }
        }
 
        public void update(String[] report){
 
            for (String s : report) {
                // 신고자, 신고당한사람을 공백으로 분리시킨 후 상태맵에 기록
                String[] log = s.split(" "); //log[0]:subject, log[1]:userName
                state.get(log[1]).add(log[0]);
            }
 
            state.values().stream().filter(reportedSet->reportedSet.size()>=criteria) // 기준치 이상을 필터링
                    .forEach(reportedSet->reportedSet.forEach(name->response.replace(name,response.get(name)+1)));
        }
 
        public int[] getResponse(){
            return response.values().stream().mapToInt(Integer::valueOf).toArray();
        }
 
    }
}