[프로그래머스] lv1 같은 숫자는 싫어 / 파이썬, 고득점kit

같은 숫자는 싫어

 

주의점!

• 입력된 arr에 맞는 순서대로 숫자를 리턴해야 한다
• 그래서 파이썬 set함수를 사용해서는 문제를 해결할 수 없다(set함수는 순서가 랜덤이니까)

 

해결방법

• 변수를 만들어 arr의 첫번째 원소 즉 arr[0]을 입력받는다
• 반복문을 돌리면서 다음 원소가 변수와 일치하지 않으면 answer에 추가, 변수 갱신

 

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