[프로그래머스 고득점 알고리즘kit] 같은 숫자는 싫어

chxxrin·2024년 4월 9일
0

문제 설명

배열 arr가 주어집니다. 배열 arr의 각 원소는 숫자 0부터 9까지로 이루어져 있습니다. 이때, 배열 arr에서 연속적으로 나타나는 숫자는 하나만 남기고 전부 제거하려고 합니다. 단, 제거된 후 남은 수들을 반환할 때는 배열 arr의 원소들의 순서를 유지해야 합니다. 예를 들면,

arr = [1, 1, 3, 3, 0, 1, 1] 이면 [1, 3, 0, 1] 을 return 합니다.
arr = [4, 4, 4, 3, 3] 이면 [4, 3] 을 return 합니다.
배열 arr에서 연속적으로 나타나는 숫자는 제거하고 남은 수들을 return 하는 solution 함수를 완성해 주세요.

제한사항

배열 arr의 크기 : 1,000,000 이하의 자연수
배열 arr의 원소의 크기 : 0보다 크거나 같고 9보다 작거나 같은 정수

입출력 예

arr answer
[1,1,3,3,0,1,1][1,3,0,1]
[4,4,4,3,3][4,3]
입출력 예 설명
입출력 예 #1,2
문제의 예시와 같습니다.

코드1 -> 추천

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

코드1 설명

  • 주어진 배열 arr를 arr의 길이만큼 반복(range(len(arr))하면서 이전 요소와 현재 요소를 비교하여 중복을 제거합니다.
  • result 리스트에 요소를 추가할 때 이전 요소와 현재 요소가 같지 않은 경우에만 추가합니다.
  • 이전 요소와 현재 요소가 같은 경우에는 추가하지 않습니다.
  • 첫 번째 요소는 항상 추가됩니다. i==0

코드2

def solution(arr):
    result = []
    for num in arr:
        if not result or result[-1] != num:
            result.append(num)
    return result

코드2 설명

  • 반복문에서 인덱스를 사용하는 대신에 for 루프에서 arr의 각 요소에 대해 직접 반복합니다.
  • 이전에 사용한 요소들을 저장하는 대신에 result 리스트의 마지막 요소와 현재 요소를 비교합니다.

코드3

def solution(arr):
    result = []
    prv = None
    for cur in arr:
        if prv != cur:
            result.append(cur)
        prv = cur
    return result

코드3 설명

  • 이전 요소를 prv 변수에 저장하여 사용합니다.
  • 현재 요소와 이전 요소를 비교하여 중복을 제거합니다.
  • prv 변수를 사용하여 직전 요소를 추적합니다.
  • prv 를 cur로 업데이트를 해줍니다.

주의할 점

  • 순서가 있기 때문에 집합 set 자료구조를 사용하면 안되고, 이전 요소와 현재 요소 비교를 반복하면 된다!

0개의 댓글