잡다한 배똥월드

import java.util.*;

class Solution {
    public int[] solution(int[] lottos, int[] win_nums) {
        int[] answer = new int[2];
        int result = 0;
        int zero = 0;
        for (int i = 0; i < lottos.length; i++) {
            if (lottos[i] != 0) {
                for (int j = 0; j < win_nums.length; j++) {
                    if (lottos[i] == win_nums[j]) {
                        result++;
                        break;
                    }
                }
            } else {
                zero++;
            }
        }        
        
        answer[0] = (result + zero == 0) ? 6 : 6 - (result + zero) + 1;
        answer[1] = (result == 0) ? 6 : 6 - result + 1;

        return answer;
    }
}

두번째 푸는 문제인데, 처음 풀었을 때의 코드가 더 깔끔한 것 같다..

일단 lottos에 0이 몇 개인지, win_nums와 같은 숫자가 몇 개인지 파악했고,

그걸 기준으로 순위를 계산했습니다.

이 문제를 풀면서 알게 된 점은 

int[] 즉, int 배열은 indexOf나 contains를 지원하지 않는다는 점이었다..!!!

그래서 for문으로 숫자가 있는지 없는지를 파악했는데, 그냥 ArrayList로 변환 후에 풀껄 그랬나.. 생각도 했다..

class Solution {
    public int[] solution(int[] lottos, int[] win_nums) {

        int bingo = 0;
        int lose = 0;

        for (int i = 0; i < lottos.length; i++) {
            if (lottos[i] != 0) {
                for (int j = 0; j < win_nums.length; j++) {
                    if (lottos[i] == win_nums[j]) {
                        bingo++;
                        break;
                    }
                }
            } else {
                lose++;
            }
        }

        int[] data = new int[]{6, 6, 5, 4, 3, 2, 1};

        return new int[]{data[bingo + lose], data[bingo]};
    }
}

.

.

.

import java.util.*;

class Solution {
    public int[] solution(String[] id_list, String[] report, int k) {
        int[] answer = new int[id_list.length];
        
        //유저 - 유저가 신고한 사람
        HashMap<String, ArrayList<String>> arr1 = new HashMap<>();

        //유저 - 유저를 신고한 사람
        HashMap<String, ArrayList<String>> arr2 = new HashMap<>();
        HashMap<String, Integer> arr3 = new HashMap<>();
        
        for (int i = 0; i < report.length; i++) {
            String[] temp = report[i].split(" ");
            
            ArrayList<String> list1 = arr1.getOrDefault(temp[0], new ArrayList<>());
            ArrayList<String> list2 = arr2.getOrDefault(temp[1], new ArrayList<>());
            
            if (list1.indexOf(temp[1]) == -1) {
                list1.add(temp[1]);
            }
            
            if (list2.indexOf(temp[0]) == -1) {
                list2.add(temp[0]);
            }
            
            arr1.put(temp[0], list1);
            arr2.put(temp[1], list2);
        }
        
        for (int i = 0; i < id_list.length; i++) {
            ArrayList<String> temp = arr2.getOrDefault(id_list[i], new ArrayList<>());
            arr3.put(id_list[i], temp.size());
        }
        
        for (int i = 0; i < id_list.length; i++) {
            ArrayList<String> temp = arr1.getOrDefault(id_list[i], new ArrayList<>());
            
            for (int j = 0; j < temp.size(); j++) {
                if (arr3.getOrDefault(temp.get(j), 0) >= k) {
                    answer[i]++;
                }
            }
        }
        
        return answer;
    }
}

해시맵 arr1은 유저가 신고한 유저 리스트

해시맵 arr2는 해당 유저를 신고한 유저 리스트

해시맵 arr3은 해당 유저가 신고당한 수 (배열로 대체 가능할 것 같음)

그래서 report 대로 for문을 돌려서 arr1과 arr2를 채워나갑니다.

만약에 중복이라면 추가하지 않고, 중복이 아니라면 추가하는 식으로...

그리고 나서 arr3을 채우는데 이건 id_list로 for문을 돌린다.

id_list에서 하나씩 꺼내서 arr1의 해당 유저가 신고한 리스트를 가져오고

리스트의 유저들이 몇 번 신고를 당했는지 파악하기 위해 arr3에서 값 가져와서

만약에 k값보다 크거나 같으면 answer++, 아니면 그냥 넘어가는 식으로 작성 후 answer 리턴