프로그래머스 Lv.0 배열의 원소 삭제하기

 

<Solution>

class Solution {
    public int[] solution(int[] arr, int[] delete_list) {
        // arr의 원소 중 delete_list의 원소와 같다면 숫자 0으로 바꿔주고 count
        // answer.lengt = arr.length - count
        int count = 0;
        for(int i = 0; i < arr.length; i++) {
            for(int j = 0; j < delete_list.length; j++) {
                if(arr[i] == delete_list[j]) {
                    arr[i] = 0;
                    count++;
                } 
            }
        }

        // answer에 넣기
        int[] answer = new int [arr.length-count];
        int idx = 0;
        for(int i = 0; i < arr.length; i++) {
            if (arr[i] != 0) {
                answer[idx] = arr[i];
                idx++;
            } 
        }
        return answer;
    }
}