[프로그래머스] 같은 숫자는 싫어 - JAVA

문제

https://school.programmers.co.kr/learn/courses/30/lessons/12906?language=java 

프로그래머스

 

코드

import java.util.*;

public class Solution {
    public int[] solution(int []arr) {
        Stack<Integer> stack = new Stack<>();
        
        for(int i=0;i<arr.length;i++){
            if(stack.empty())
                stack.push(arr[i]);
            else if(stack.peek() != arr[i]) {
                stack.push(arr[i]);
            }

                
        }
        
        int answer [] = new int [stack.size()];
        
        int n = stack.size();
        int i = 0;
        while(!stack.empty()){
            int num = stack.pop();
            // System.out.println(num);

            answer[ n -i -1] = num;
            i++;
        }
        

        return answer;
    }
}