[알고리즘 프로그래머스] 같은 숫자는 싫어 (LEVEL 1)
(문제) 같은 숫자는 싫어
배열 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에서 연속적으로 나타나는 숫자는 제거하고 남은 수들을 return 하는 solution 함수를 완성
테스트케이스
입력예제 | 출력예제 |
---|---|
[1,1,3,3,0,1,1] | [1,3,0,1] |
[4,4,4,3,3] | [4,3] |
[5,4,3,2,1] | [5,4,3,2,1] |
해결방법
방법 1
리스트의 반복되지 않는 것
만 넣는 리스트를 생성
리스트를 순회하면서 만약에 리스트의 반복되지 않는 것
리스트 마지막 요소가 같으면 그냥 넘어가고 같지 않으면 리스트의 반복되지 않는 것
리스트에 삽입을 한다.
방법 2
입력으로 들어오는 리스트의 반복문으로 순회하면서 (앞, 뒤) 이런 식으로 비교하여 만약 다를 경우 앞에 있는 것을 삽입한다.
마지막에는 맨 뒤에 있는 값을 삽입한다.
코드 방법 1
def solution(x):
answer = []
for i in range(len(x)):
if answer[-1:] != [x[i]]:
answer.append(x[i])
return answer
코드 방법 2 (파이썬 nic 1)
def solution(x):
answer = []
for i in range(len(x)-1):
# 마지막일 경우 그냥 끝낸다
# 다르면 앞에 있는 것을 추가
if x[i] != x[i+1]:
answer.append(x[i])
return answer + x[-1:]
코드 방법 2 (파이썬 nic 2)
def solution(x):
# 연속적으로 나타나는 숫자는 하나만 남기고 전부 제거
# zip으로 앞 뒤 끊어서 가져온 후, 계속 검사
# 마지막의 값은 무시가 되서 해당 마지막 부분을 합해서 가져온다.
return [i for i,j in zip(x, x[1:]) if j != i] + x[-1:]
댓글남기기