본문 바로가기
Coding Test

231026 같은 숫자는 싫어 - Level 1

by mmm- 2023. 10. 26.

문제 설명

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

 

입출력 예

 

입출력 예 설명

입출력 예 #1,2
문제의 예시와 같습니다.


코드

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

 

풀이

arr의 길이-1만큼 for문을 돌려 arr의 i번째 자리의 수와 i+1번째 자리의 수를 비교해준다. 그 값이 다르면 arr[i]을 answer에 추가하는데, for문을 벗어났을 때 arr리스트의 마지막에 존재하는 값은 len(arr)-1만큼만 for문을 돌렸기 때문에 추가되지 않았으므로 추가해준다.


다른 코드와 풀이

아래의 코드는 다른 사람들이 작성한 것들로, 내 코드와 비교, 참고하기 위해 가져왔다.

def no_continuous(s):
    # 함수를 완성하세요
    result = []
    for c in s:
        if len(result) == 0 or result[-1] != c:
            result.append(c)

    return result

result에 아무것도 추가되지 않아 길이가 0이거나 result에 마지막으로 추가된 값과 s에 있는 값이 다르면 result에 추가해준다.