CoTe/[programmers]1

큐같은 숫자는 싫어(스택, 큐)

TheSole 2024. 1. 31. 15:36

https://school.programmers.co.kr/learn/courses/30/lessons/12906

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

문제 설명

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

def solution(arr):
    s = arr
    a = []
    for i in range(len(s)):
        if i == 0:
            a.append(s[i])
        elif arr[i] != s[i-1]:
            a.append(s[i])
        
    return a
  • 문제 접근 자체가 잘못됨. 지운다기 보단 비교하고 추가함.
  • 이전까지는 두 개 비교 하기위해서 for문 돌리고 index 초과 났는데 위 방법을 보면
    • 처음 인덱스가 0일 때 그냥 리스트에 넣어버리고 이후에 인덱스"1"로 시작하는 값과 이전 리스트에 넣었던 이전값 인덱스 "0"과 계속 비교하고 같지 않으면 추가하는 느낌임.
def solution(arr):
    
    a = []
    for i in arr:        
        if a[-1:] == [i]: continue
        a.append(i)
         
    return a
  • 창의력 그 잡채
  • 빈 리스트를 만들어 주고 현재 배열이랑 비교를 바로함
    • 여기서 슬라이싱을 "if a[-1:]"을 하면 항상 맨마지막꺼랑 반복문 순서랑 비교하게 되는데 여기서 핵심은 빈리스트에 아무것도 없으니 바로 i가 추가되고 이후에는 만약 a 값의 인덱스 마지막이랑 비교해서 같으면 continue 문으로 스킵해버림
      • 주의할점
        • a.append() if문 밖으로 보내는 것 보내지 않으면 조건문에서 부터 씹힘
        • a[-1:] 이 개별 리스트로 들어오기 떄문에 똑같이  [i]를 만들어줘야함