[프로그래머스] 택배상자

[프로그래머스] 택배상자 바로가기

아이디어

• stack쓰는 건 알았는데 구현을 못했다... 🥲

시간 복잡도

• O(n**2)

코드

def solution(order):
    answer = 0
    box = 1
    stack = []
    
    while box != len(order) + 1:
        # 무족권 보조 컨테이너에 넣기
        stack.append(box)
        
        # 보조 컨테이너에 있는게 지금 실어야될 상자인지 체크(안같을 때까지)
        while stack[-1] == order[answer]:
            answer += 1
            stack.pop()
            
            # 보조 컨테이너가 비면 종료
            if not stack:
                break
        
        box += 1
    
    return answer