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

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

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

 

해당 문제 조건

• 각 유저는 한번에 한명 유저만 신고가능
• 동일 유저 신고는 못하고 자기자신 신고불가.
• 2 ≤ id_list ≤ 1,000 소문자로만 이루어져있다.
• 1 ≤ id_list 원소의길이 ≤ 10
• 1≤ report ≤ 21 {이용자id 신고한id} 로 구분자 스페이스바
• 1≤k≤200

시간복잡도

최악의 경우 report_id * id_list 해도 O(2억)정도 되려나 그래서 제한시간 10초길래  일단은 for문으로 작성해봤다.

 

 

풀이방법

• 동일 유저 신고를 못한다고 조건에 나열했으니 set 자료구조로 report배열을 받아주고
• 신고당한 id 기준으로 신고한 유저 리스트 나열   {신고당한 id: 신고 유저1, 유저2}  (Map)
• 유저리스트 안에 id 반복문을 돌려서 id_list 인덱스 위치에 카운트 +1(신고 횟수)
• retrun 해줌.

끝!.

 

---

구현

import java.util.*;
class Solution {
    public static int[] solution(String[] id_list, String[] report, int k) {
        int[] answer = new int[id_list.length];
        ArrayList<String> userId = new ArrayList<>(Arrays.asList(id_list));
        Map<String, ArrayList<String>> reported = new HashMap<String, ArrayList<String>>();
        Set<String> set = new HashSet<String>(Arrays.asList(report));

        for (String s : set) {
            String[] s1 = s.split(" ");
            ArrayList<String> strings = new ArrayList<String>();
            if (reported.containsKey(s1[1])) {
                strings = reported.get(s1[1]);
                strings.add(s1[0]);
            } else {
                strings.add(s1[0]);
            }
            reported.put(s1[1], strings);
        }

        reported.forEach((key, value) -> {
            if (value.size() >= k) {
                for (String s : value) {
                    int count = userId.indexOf(s);
                    answer[count] += 1;
                }
            }

        });

        return answer;
    }
}

 

요런 느낌으로 풀었습니다.

 

피드백 주시면 감사합니다.