[java 프로그래머스] LV.1 신고결과받기

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

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

 
import java.util.List;
import java.util.ArrayList;
 
class Solution {
    public int[] solution(String[] id_list, String[] report, int k) {
 
        String attack[]=new String[id_list.length];
        //내가 신고한것을 확인하기 위한 배열
        List<String>alarm[]=new ArrayList[id_list.length];
        for(int i=0;i<id_list.length;i++){
            alarm[i]=new ArrayList<String>();
        }
 
        for(int i=0;i< report.length;i++){
            String str[]=report[i].split(" ");
            for(int j=0;j< id_list.length;j++){
                if(id_list[j].equals(str[0])){
                    if(!alarm[j].contains(str[1])){
                        alarm[j].add(str[1]);
                    }
 
                }
            }
        }
 
 
        //신고당한 횟수
        List<String> damage[]=new ArrayList[id_list.length+1];
        for(int i=0;i<damage.length;i++){
            damage[i]=new ArrayList<String>();
        }
        for(int i=0;i<report.length;i++){
            String []str=report[i].split(" ");
            //공격쪽 str[0] 피해입은 쪽 str[1]
            for(int j=0;j<id_list.length;j++){
                if(str[1].equals(id_list[j])&&!damage[j].contains(str[0])){
                    damage[j].add(str[0]); //신고한 칭구를 안에 넣는다.
                }
            }
 
        }
 
 
        ArrayList<String>pause=new ArrayList<String>();
        for(int i=0;i<damage.length;i++){
            if(damage[i].size()>=k){
                pause.add(id_list[i]);
            }
        }
        int ans[]=new int[id_list.length];
        for(int i=0;i<alarm.length;i++){
            int cnt=0;
            for(int j=0;j<alarm[i].size();j++){
                for(int s=0;s< pause.size();s++){
                    if(alarm[i].get(j).equals(pause.get(s))){
                        cnt++;
                    }
                }
            }
            ans[i]=cnt;
        }
        return ans;
    }
}