프로그래머스/파이썬

같은 숫자는 싫어_알고리즘(스택/큐)

싱싱한복초이 2025. 2. 1. 10:18

https://school.programmers.co.kr/learn/courses/30/lessons/12906?language=python3

 

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

[문제 설명]

배열 arr가 주어집니다. 배열 arr의 각 원소는 숫자 0부터 9까지로 이루어져 있습니다. 이때, 배열 arr에서 연속적으로 나타나는 숫자는 하나만 남기고 전부 제거하려고 합니다. 단, 제거된 후 남은 수들을 반환할 때는 배열 arr의 원소들의 순서를 유지해야 합니다. 예를 들면,

  • arr = [1, 1, 3, 3, 0, 1, 1] 이면 [1, 3, 0, 1] 을 return 합니다.
  • arr = [4, 4, 4, 3, 3] 이면 [4, 3] 을 return 합니다.

배열 arr에서 연속적으로 나타나는 숫자는 제거하고 남은 수들을 return 하는 solution 함수를 완성해 주세요.

 

제한사항

  • 배열 arr의 크기 : 1,000,000 이하의 자연수
  • 배열 arr의 원소의 크기 : 0보다 크거나 같고 9보다 작거나 같은 정수

[나의 풀이]

def solution(arr):
    answer = []
    answer.append(arr[0])
    for i in range(1,len(arr)):
        if arr[i-1] != arr[i]:
            answer.append(arr[i])
    return answer

 

[다른 풀이]

1)

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

 

2)

def solution(arr):
    result = []
    for c in arr:
        if len(result) == 0 or result[-1] != c:
            result.append(c)

    return result

 

3)

def solution(arr):
    return [arr[i] for i in range(len(arr)) if [arr[i]] != arr[i+1:i+2]]

 

☑️여기서 질문

def solution(arr):
    return [arr[i] for i in range(len(arr)) if arr[i] != arr[i+1]]

이 코드는 왜 IndexError: list index out of range 라고 출력이 되는가? 

 

=> 이 오류는 배열의 마지막 인덱스에 접근할 때 발생한다.

 

(해결방법)