[프로그래머스] 같은 숫자는 싫어 (Python)

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

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

문제 설명

· 배열 arr에서 연속으로 나타나는 숫자를 제거하고 반환

접근 방법

· arr의 원소들을 하나씩 stack의 마지막 값과 같지 않다면 넣지 않음

· 첫 시행 때 스택이 비어 있으므로 주의

알고리즘 설계 및 구현

def solution(arr):
    stack = []
    for c in arr:
        if not stack or c != stack[-1]:
            stack.append(c)
    return stack

결과