문제풀이/일일연습문제

[stack/queue]99클럽 코테 스터디 7일차 TIL + 프로그래머스/1/12906. 같은 숫자는 싫어

Mo_bi!e 2024. 7. 28. 21:49

문제 설명

배열 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,2
문제의 예시와 같습니다.

 

 

<내 코드>

def solution(arr):
    answer = []
    
    for a in arr:
        if len(answer) == 0:
            answer.append(a)
        else :
            if answer[-1] != a:
                answer.append(a)
            
    return answer
        
# set()은 안됨 -> [1,3,0,1] 중 [0,1,3] 으로 나옴
# hash() 도 안됨 -> 마찬가지
# 즉 두개 다 중복이 없고, 순서가 정렬되어서 출력됨

1. 처음에 연달아 오는것을 똑같이 한다고 생각해서, 중복제거용으로 hash, set 을 생각함

사실 문제 조건은 다른 부분은 중복이 가능하고, 들어온 순서대로 그대로 있어야함

 

2. 이후 answer 리스트의 길이가 0 인 조건에 대해서 and 와 or 헷갈림

아래 모범사례에서 잘 표현을 해줌

 

 

 

<모범사례>

- index 를 -1로만한 경우

def solution(arr):
    answer = []
    
    # Iterate through the array and add non-duplicate consecutive elements
    for a in arr:
        if not answer or answer[-1] != a:
            answer.append(a)
    
    return answer

 

- 리스트 슬라이싱을 한 경우

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

 

이 두가지를 살펴보면

a[-1] 과 a[-1:] 의 차이를 알 수있음 전자는 요소 하나만 리턴을 하지만, 후자는 리스트를 리턴함