[프로그래머스] 파이썬 코딩테스트 연습 LV1 (같은 숫자는 싫어)

1. 같은 숫자는 싫어

 

문제

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

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

 

내 풀이

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

elif i == 0:: 만약 i가 0이면(첫 번째 원소), 바로 이전 원소가 없기 때문에 answer 리스트에 추가한다.