프로그래머스 - LV2. 택배상자

문제

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

프로그래머스

 

코드

def solution(order):
    answer = 0
    package = list(range(1, len(order)+1))
    stack = []
    
    for o in order:
        if o in package:
            index = package.index(o)
            stack += package[:index]
            answer += 1
            package = package[index+1:]
        elif o == stack[-1]:
            answer += 1
            stack.pop()
        else:
            break
    
    return answer

테스트 케이스 6개는 맞고 4개는 시간 초과!