프로그래머스 - java 빈 배열에 추가, 삭제하기

 

 💡 flag 값에 따라 arr 에 있는 값들을 추가하고 제거해야한다.

 

flag 길이와 arr 길이는 동일하다.

flag 를 1차로 돌려주면서 flag 값이 T & F 인지 판단해준다.

1. 참이면 arr[i] 를 arr[i] *2 번 반복해준다.

2. 거짓이면 arr[i] 만큼 지운다.  (지울때는 뒤에서부터 remove)

 

 

 💬 배열은 switch 값에 사용 불가 : flag 배열이 boolean 배열로 선언되어서, switch문 대신 if-else 문을 사용해줘야한다.

 💬 arr.size() :: 배열의 크기를 반환한다.

import java.util.ArrayList;

class Solution {
    public ArrayList<Integer> solution(int[] arr, boolean[] flag) {
        ArrayList<Integer> answer = new ArrayList<Integer>();
        
       
        for (int i = 0; i < flag.length; i++){
                if (flag[i] == true){  //flag 가 참이면 arr[i] * 2 만큼 출력
                    for (int j = 0; j < arr[i]*2; j++){
                        answer.add(arr[i]);
                    }
                }else{ // 거짓이면 arr[i] 
                    for (int j = 0; j < arr[i]; j++){
                     answer.remove(answer.size()-1);
                    }
                }

        }//for
        
        return answer;
    }
}