프로그래머스 배열 만들기 6 (JAVA)

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

프로그래머스

문제가 잘못되어있습니다.

 

만약 stk이 빈 배열이라면 arr[i]를 stk에 추가합니다 -> 만약 stk이 빈 배열이라면 arr[i]를 stk에 추가하고 i를 1 증가시킵니다. 

위의 조건으로 풀어야 문제를 풀 수 있습니다.

 

import java.util.*;

class Solution {
    public int[] solution(int[] arr) {
        int[] answer = cal(arr);
        return answer;
    }
    private static int[] cal(int[] arr){
        Stack<Integer> stack = new Stack<>();
        int index = 0;
        while(index < arr.length){
            if(stack.isEmpty()){
                stack.push(arr[index]);
                index++;
                continue;
            }
            if(stack.peek() == arr[index]){
                stack.pop();
                index++;
            }else{
                stack.push(arr[index]);
                index++;
            }
        }
        if(stack.size() == 0){
            return new int[]{-1};
        }
        int[] result = new int[stack.size()];
        for(int i = stack.size() - 1 ; i >= 0 ; i--){
            result[i] = stack.pop();
        }
        return result;
    }
}