소소한 컴퓨터 이야기

같은 숫자는 싫어

by Cori

문제

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

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

리스트 탐색을 수월하게 하기 위해 리스트 뒤에 큰 값을 하나 추가하였다. (arr[len(arr)]을 사용하기 위함)  리스트 탐색을 하며, i 값이 i + 1값과 같지 않은 경우에만 새로운 리스트에 저장하고, 이를 반환하였다.

 

2. Others

def no_continuous(s):
    a = []
    for i in s:
        if a[-1:] == [i]: continue
        a.append(i)
    return a

a[-1: ] -> 리스트 a의 맨 마지막에 있는 요소를 리스트로 만듬.. 리스트 a에 중복값이 있으면 continue를 활용하여 넘어가고, 중복값이 없을 경우 요소를 추가하여 반환하였다. 확실히 파이썬은 리스트 슬라이싱을 활용하면 코드가 간결하고, 보기 좋아지는 듯.. ! 또 하나 배워간다. 

 

* if a[-1:] == [i] 대신 if a[-1] == i 로 사용하면, index out of range 오류가 발생함 .. a[-1:]과 a[-1]의 차이 ? 

'CS > Coding Test' 카테고리의 다른 글

2016년  (0) 2021.08.22
가운데 글자 가져오기  (0) 2021.08.22
나누어 떨어지는 숫자 배열  (0) 2021.08.22
두 정수 사이의 합  (0) 2021.08.21
문자열 내 p와 y의 개수  (0) 2021.08.21

블로그의 정보

코딩하는 오리

Cori

활동하기