코테 공부/python

[프로그래머스] 같은 숫자는 싫어 -stack

sweet-po 2023. 6. 28. 15:42

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

 

내풀이

길이 0일 경우를 따로 처리해주지 않으면 index가 초과해버리길래

이렇게 분화해주었지만

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

 

타풀이

에서 기깔나게 리스트 슬라이싱 해서 리스트로 비교하여 분화 없이 해결하였다

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

 

+ 이렇게 해보니 인덱스 초과 없이 빈리스트로 반환하더라

a = []
a = a[-2:]
print(a)
#[]