[프로그래머스 lv 1] 같은 숫자는 싫어.py

<문제 설명>

https://programmers.co.kr/learn/courses/30/lessons/12906

코딩테스트 연습 - 같은 숫자는 싫어

배열에서 연속으로 중복 숫자가 나오면 이를 제외시킨 배열을 리턴하는 문제 이다.

 

입출력의 예는 다음과 같다.

arr	answer
[1,1,3,3,0,1,1]	[1,3,0,1]
[4,4,4,3,3]	[4,3]

 

<문제 해결 아이디어>

tmp라는 변수에 이전 숫자를 저장하게 하고

tmp와 지금 숫자가 같지 않다면 리스트에 저장한 후 tmp를 지금 숫자로 바꿔주었고

지금 숫자와 같다면 리스트에 저장하지 않고 tmp를 지금 숫자로 바꿔주었다.

tmp는 배열의 첫 숫자로 초기화 한 후 이 역시 맨 처음에 리턴 리스트에 추가했다.

 

<문제 풀이 코드>

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

 

채점 결과

ps. 처음에 문제 잘 못 보고 set으로 푸는 문제로 착각하여 풀었다가 코드 수정했다. 문제를 제대로 읽자

 

제안해주신 수정 코드

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