[코딩 기초 트레이닝] 빈 배열에 추가, 삭제하기

문제 :

https://school.programmers.co.kr/learn/courses/30/lessons/181860

프로그래머스

 

해결 :

import java.util.*;

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

 

다른 사람 풀이 :

import java.util.*;

class Solution {
    public int[] solution(int[] arr, boolean[] flag) {
        Stack<Integer> stack = new Stack<>();

        for (int i = 0; i < flag.length; i++) {
            if (flag[i]) {
                for (int j = 0; j < arr[i] * 2; j++) stack.push(arr[i]);
            } else {
                for (int j = 0; j < arr[i]; j++) stack.pop();
            }
        }

        return stack.stream().mapToInt(Integer::intValue).toArray();
    }
}