프로그래머스 - 같은 숫자는 싫어 [python]

난이도 : level 1
유형 : 스택
출처 : https://school.programmers.co.kr/learn/courses/30/lessons/12906

문제 파악

0부터 9로 이루어져있는 배열 arr 를 입력 받습니다.
[1,1,3,3,0,0,1,1] 을 입력받았다면
우리가 구해야할 값은 연속된 숫자들을 제거한 새로운 배열입니다.
-> [1,3,0,1]

[4,4,4,3,3] 을 입력받을 시
-> [4,3]

해답 및 풀이

def solution(arr):
    
    answer = []
    for i in range(len(arr)):
        if i == 0 : # 첫 배열의 값은 answer에 바로 넣어줍니다. i-1을 해줄경우 -1이 되어 오류가 발생하기 때문입니다.
            answer.append(arr[i])
        elif arr[i] != arr[i-1]: # i=1 부터는 arr에서 그 이전 값과 비교해서 같지 않을때만 answer에 append 해줍니다.
            answer.append(arr[i])
    return answer