하루살이 개발자

[프로그래머스] Level1 - 신고결과 받기(Java) 본문

코딩테스트

[프로그래머스] Level1 - 신고결과 받기(Java)

하루살이 2022. 8. 5. 17:30

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

import java.util.*;
class Solution {
    public int[] solution(String[] id_list, String[] report, int k) {
        int[] answer = new int[id_list.length]; // 초기화
        ArrayList<HashSet<String>> reported = new ArrayList<HashSet<String>>(); 
      
        HashMap<String, Integer> idIdx = new HashMap<String, Integer>(); // 초기화 

        for (int i = 0; i < id_list.length; i++) {
            idIdx.put(id_list[i], i); // 이름, 0~3숫자
            /*
            {
             muzi : 0
             frodo : 1
             apeach : 2
             neo : 3
            }
            */
            reported.add(i, new HashSet<String>());
        }

        for (String names : report) {
            String reportPerson = names.split(" ")[0]; // "1,2 "에서 1
            String reportedPerson = names.split(" ")[1]; // "1,2 "에서 2

            System.out.println("reportPerson:" + reportPerson);
            System.out.println("reportedPerson:" + reportedPerson);

            reported.get(idIdx.get(reportedPerson)).add(reportPerson); //name 찾아서 넣기
            System.out.println("reported:" + reported);
        }

        for (HashSet<String> set : reported) {
            if (set.size() >= k) {
                for (String name : set) {
                    answer[idIdx.get(name)]++;
                }
            }
        }
        return answer;
    }
}