codes/programmers

12906 같은 숫자는 싫어

카제xd 2024. 12. 12. 02:26

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

 

def solution(arr):
    
    stack = []
    for ar in arr:
        if stack:
            if stack[-1] == ar:
                continue
        stack.append(ar)
            
    return stack

 

 

[개선할 점]

- stack[-1:] 의 형태로 쓰면 빈 배열에 써도 된다. 그러면 if stack을 없앨 수 있다.

- 조건문 두번을 if not stack or stack[-1] != ar 로 대체할 수도 있다.