[Algorithm] 스택/큐 같은 숫자는 싫어

문제 링크

프로그래머스

 

내가 제출한 코드

import java.util.*;

public class Solution {
    public int[] solution(int []arr) {
        int[] answer = {};
        Stack<Integer> stack = new Stack<>();
        
        for(int number : arr){
            if(stack.size()!=0 && stack.peek() == number) continue;                
            stack.push(number);
        }
        answer = new int[stack.size()];
        for(int i = answer.length -1; i >= 0; i--){
            answer[i] = stack.pop();
        }
        return answer;
    }
}

Stack을 이용해 하나 씩 push하고 peek를 통해 가장 위의 값과 들어오는 값이 같은지 비교하도록 했다.

결과