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

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

프로그래머스

 

 

 

 

 

 

정답 

 

import java.util.ArrayList;

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;
    }
}

 

 

다른 사람의 풀이 

 

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();
    }
}