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

3166 단어 algorithmalgorithm

Algorithm Problem with Python — 10day


문제 설명 📖

배열 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보다 작거나 같은 정수

입출력 예


문제 이해 🔑

배열 내 연속적으로 존재하는 중복된 요소를 제거하는 문제입니다.
배열을 순회해야하며 각 인덱스의 앞 또는 뒤를 비교를 해야합니다.
0번 째 인덱스를 answer배열에 미리 넣어두고 1번 째와 0번 째를 비교하면서 시작하겠습니다.
그럼 즉 i-1(0번 째)와 i(1번 째)의 비교로 배열의 범위를 넘지 않으면서 끝까지 비교할 수 있습니다.


수도 코드 ✍️

  1. 반복문을 통해 arr을 순회합니다.
  2. 배열의 범위 안에서 비교하기 위해서 0번 째는 무조건 넣어줍니다.
    3-1. i번 째와 i-1의 값을 비교하여 동일하면 무시합니다.
    3-2. 같지 않다면 미리 만들어 둔 배열에 값을 넣습니다.

코드 작성 ⌨️

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

정리 😄

배열의 범위를 넘어서지 않기 위해서 비교 대상의 기준을 잡아야 했습니다.
i와 i+1를 비교하면 배열의 마지막 요소에 다다랐을 때 배열의 범위를 넘어서기 때문에 에러가 발생됩니다.
그래서 0번 째를 미리 answer배열에 넣어두고 i와 i-1을 비교하는 방식을 택했습니다.

좋은 웹페이지 즐겨찾기