프로그래머스 Lv.0 빈 배열에 추가, 삭제하기

 

<Solution>

import java.util.*;

class Solution {
    public int[] solution(int[] arr, boolean[] flag) {
        ArrayList<Integer> list = new ArrayList<> ();
        for(int i = 0; i < flag.length; i++) {
            if(flag[i]) {
                for(int j = 0; j < arr[i] * 2; j++) {
                    list.add(arr[i]);
                }
            }else {
                for(int j = 0; j < arr[i]; j++) {
                    list.remove(list.size()-1);
                }
            }
        }
        
        int answer[] = new int [list.size()];
        for(int i = 0; i < list.size(); i++) {
            answer[i] = list.get(i);
        }
        return answer;
    }
}