[Programmers/알고리즘 고득점 KIT] 같은 숫자는 싫어

[Programmers] 같은 숫자는 싫어

Level 1 | #스택/큐

문제 바로가기 🔗

1. 문제 요약

배열 arr에서 연속적으로 나타나는 숫자는 제거하고 남은 수들을 반환하는 문제이다.

단, 제거된 후 남은 수들을 반환할 때는 원래의 순서를 유지해야 한다.

2. 접근 방식 및 포인트

배열의 원소를 하나씩 순회하며 가장 최근에 추가된 숫자와 비교하여 중복 여부를 판단한다.

• ✅ 스택 활용: 새로운 배열(스택)을 만들고, 마지막 요소(stack[-1])가 현재 숫자와 다를 때만 추가한다.
• ✅ 예외 처리: 스택이 비어있는 초기 상태에서는 비교할 대상이 없으므로 무조건 숫자를 추가해야 한다.

3. 구현 코드 (Python)

def solution(arr):
    stack = []
    for num in arr:
        # 스택이 비어있거나, 마지막 요소가 현재 숫자와 다르면 추가
        if not stack or stack[-1] != num:
            stack.append(num)
            
    return stack

📌 논리 구조: not stack 조건이 먼저 오기 때문에, 스택이 비어있을 때 발생하는 IndexError를 방지할 수 있다.

4. 🤔 if not stack: vs if len(stack) == 0:

둘 다 빈 리스트를 체크하는 코드이지만, 파이썬에서는 관용적으로 if not stack 방식을 권장한다.

• 가독성: 코드가 더 간결하고 "파이썬스럽다(Pythonic)".
• 성능: 미세한 차이지만 함수 호출(len()) 과정이 없어 전자가 더 빠르다.