[Programmers] 같은 숫자는 싫어

시작하기 전에

프로그래머스 알고리즘 고득점 Kit에서 스택/큐 단원의 첫 번째 문제입니다.

---

I. Description

---

II. Code

def solution(arr):
    stack = []
    
    for elm in arr:
        # 스택이 비어있는 경우에는 무조건 추가
        if not stack:
            stack.append(elm)
        # 배열의 원소가 스택의 탑과 값이 다른 경우에만 추가
        if stack[-1] != elm:
            stack.append(elm)
            
    return stack

# Test cases
if __name__ == "__main__":
    arr_list = [[1, 1, 3, 3, 0, 1, 1], [4, 4, 4, 3, 3]]
    for arr in arr_list:
        result = solution(arr)
        print(result)

 

스택을 활용하여 간단하게 풀어낼 수 있다.