알고리즘/Python

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

짱호 2020. 7. 26. 16:20
반응형

문제

풀이

arr 배열안에 연속적으로 나타나는 숫자를 하나만 남기고 제거하는 문제이다.

ans라는 배열에 arr의 첫번째 값을 초기값으로 넣고, 이후 수를 비교한다.

ans의 -1번째 수, 즉 ans 배열의 마지막 원소를 arr[i]번째 수와 비교하여 답을 찾을 수 있다.

코드

def solution(arr):
    ans = list()
    ans.append(arr[0])
    for i in range(1, len(arr)):
        if ans[-1] != arr[i]:
            ans.append(arr[i])
        else:
            continue
    return ans
반응형