programming

프로그래머스 같은 숫자는 싫어 퀴즈 파이썬 풀이 해설


by Kitle · 2020. 07. 19.



원본 문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/12906

문제 설명

배열 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

문제의 예시와 같습니다.


[풀이]

def solution(arr):
answer = []
for idx, item in enumerate(arr):
if idx == 0:
answer.append(item)
elif arr[idx - 1] != item:
answer.append(item)
return answer

[해설]
배열 arr에서 연속적으로 나타나는 숫자를 제거하고 순서를 유지해야 합니다.
따라서 sort를 하는 경우는 1,2,1 등의 순서 유지를 할 수 없으므로 사용해서는 안됩니다. 
복제 배열을 만들고 index를 비교하여 중복되면 빼는 방식이 있겠지만, 여기서는 역으로 빼지않고, 중복되지 않은 리스트를 새로 만들어줍니다. 맨뒤에 append() 로 하나씩 붙여주면 순서도 유지되고 중복은 조건문으로 걸러주면 되니 더 쉽다고 볼 수 있습니다.
첫번째 문장은 예외처리입니다. 길이가 1 일 수도 있기때문에 이 경우는 중복과 상관없이 무조건 리스트에 추가해 줍니다. 
if idx == 0:
            answer.append(item)
길이가 0인 경우는 아무것도 안하고 빈 리스트가 아마도 리턴될 겁니다.
두번째로 index가 0이 아닌 경우 (여기서는 0보다 큰 경우로 생각하고 처리되었네요.. 좋은 문장은 아니네요)

elif arr[idx - 1] != item:
answer.append(item)
배열의 비교는 바로 앞에 값이 있어야 의미가 있습니다. 따라서 index가 0보다 큰 경우부터 실시합니다. 이전 값과 같다면 중복되므로 새 리스트에 추가할 필요가 없습니다.
이전의 값과 다르다면 다른 값이므로 append를 통해 추가해 줍니다. 맨 뒤에 추가 되므로 자연스럽게 순서가 유지됩니다.
어렵지 않게 풀 수 있겠죠?
더 간략하게 하는 방법도 많을 것입니다. 이 부분을 한번 고민해보세요 :)