[Python] 같은 숫자는 싫어

이전에 C++로 풀었던 문제를 python으로 다시 풀었다.

입력받은 리스트 arr에 대해서, 연속적으로 나타나는 숫자는 하나만 남기고 전부 제거해야 하는 문제이다.

 

arr 리스트 요소들을 저장할 변수 arr_num 과, answer 리스트에 이미 들어간 요소인 지 확인하기 위한 변수 answer_num을 선언해 준다.

 

def solution(arr):
    answer = []
    #answer 리스트에 들어갈 요소
    answer_num = None
    for i in arr:
        #arr 리스트 요소
        arr_num = i
        #answer 에 이미 들어간 요소인 지 체크
        if arr_num == answer_num:
            continue
        answer_num = arr_num
        answer.append(arr_num)
    return answer