[파이썬 코딩테스트] 같은 숫자는 싫어

ch.2·2024년 7월 22일
0

코딩 테스트

목록 보기
13/21
post-thumbnail

프로그래머스 연습 문제

배열 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 함수를 완성해 주세요.


답안 분석

answer에 arr[0]을 담아두고 이후의 것들을 비교하는 방법

def solution(arr):
    answer = [arr[0]] 
    for i in range(1, len(arr)):
        if arr[i] != arr[i-1] : 
            answer.append(arr[i])
    return answer
  • if arr[i] != arr[i-1] :
    : arr[i] 와 arr[i+1]을 비교한다면, 마지막 반복 때 마지막 인덱스인 arr[i+1]은 arr에 없는 인덱스이기 때문에 오류가 난다. 따라서 i와 i-1을 비교하는 if문을 작성한다.

  • for i in range(1, len(arr)):
    : if 문에서 i-1을 사용하기 위해 range의 범위를 1부터 len(arr)까지로 설정한다.

  • answer = [arr[0]]
    : range가 1부터 시작하므로, 첫 번째 요소인 arr[0]을 미리 answer 리스트에 담아둔다



답안 분석 2

for 문에서 answer에 arr[i]를 담고, 이후에 if 문으로 arr[i]가 연속되는 숫자인지 비교해서, 연속된다면 pop()으로 빼내는 방법

def solution(arr):
    answer = []
    for i in range(len(arr)):
        answer.append(arr[i])
        if [arr[i]] == arr[i + 1: i + 2]:
            answer.pop()
    return answer
  • if [arr[i]] == arr[i + 1: i + 2]:
    : arr[i + 1: i + 2]를 통해 슬라이싱을 하면 결과값이 리스트 안에 담기므로, arr[i]도 리스트로 감싸서 비교한다.

pop()을 사용하지 않고 간소화한 방법

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


답안 분석 3

len(answer)의 값이 0일 때 담고, answer에 담긴 이전값과 i가 같지 않다면(연속되지 않는 숫자라면) answer 변수에 담는 방법

def solution(arr):
    answer = []
    for i in arr:
        if len(answer) == 0 or answer[-1] != i:
            answer.append(i)
    return answer
  • answer 변수에 i를 처음부터 담고, 이후에 answer[-1]와 i를 비교한다.

  • 답안 분석 2번을 간소화한 방법과 비슷하지만 if 문에서 len(answer) == 0 인 것을 활용한 것이 새롭게 느껴져서 분석해보았다.



답안 분석 4

arr[-1]부터 비교해서 answer에 담는 방법.

def solution(arr):
    answer = []
    for i in arr:
        if answer[-1:] == [i]: continue
        answer.append(i)
    return answer
  • arr[-1]부터도!! 비교가 가능하다.

  • if answer[-1:] == [i]:
    : if answer[-1] == [i] 을 사용하면 answer 리스트가 비어있는 경우 인덱스 오류가 발생한다.
    따라서 [-1:]를 통해 answer 리스트의 마지막 요소만 선택한 후, 리스트 [i]와 비교하는 구문이다.

profile
데이터 분석 공부중

0개의 댓글