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

문제

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

프로그래머스

 

 

 

풀이

스택을 이용해 연속되는 숫자를 관리한다.

arr의 숫자를 스택에 저장할 때 스택의 top에 있는 숫자와 지금 저장할 숫자가 같은지 확인한다.

 

 

 

코드

python

def solution(arr):
    answer = []
    
    for n in arr:
        if not (answer and n == answer[-1]):
            answer.append(n)
    
    return answer

 

java

import java.util.Stack;
 
public class Solution {
    public int[] solution(int[] arr) {
        Stack<Integer> st = new Stack<>();
        st.add(arr[0]);
        for (int i = 1; i < arr.length; i++) {
            if (arr[i] != st.peek()) {
                st.add(arr[i]);
            }
        }
        int[] answer = new int[st.size()];
        int idx = st.size() - 1;
        while (!st.empty()) {
            answer[idx--] = st.pop();
        }
 
        return answer;
    }
}