[Python] 같은 숫자는 싫어 - 프로그래머스(Lv.1)

 

 

나의 문제 풀이

def solution(arr):
    answer = []
    i=0
    while i<len(arr)-1:
        if arr[i] != arr[i+1]:
            answer.append(arr[i])
        i+=1
    answer.append(arr[-1])
    return answer

 

먼저 결과값을 만들 빈 리스트를 만듭니다.

그리고 while문을 arr리스트 마지막 인덱스-1 까지 돌립니다. 왜냐하면 아래 if문을 돌릴 때 i+1이 있어서 리스트 범위 밖으로 돌렸다고 에러가 나기 때문입니다. 

자신의 원소와 다음 원소가 다를 경우 answer에 그 원소를 추가하는 방식으로 돌립니다. 그러나 이렇게 할 경우 마지막 인덱스 부분이 추가가 되지 않기 때문에 이를 위해 마지막 인덱스는 추가하는 방식으로 해서 답을 구했습니다.