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

문제

문제를 이해하는 게 좀 어려울 뿐이지, 문제자체는 쉽다

stack을 사용해서 풀면된다.

코드

def solution(order):
    answer = 0
    st = []
    done = False
    num = 1
    st.append(num)
    for i in order:  
        while True:
            if st and i < st[-1]:
                done = True
                break
            if st and i == st[-1]:
                answer += 1
                st.pop()
                break
            num += 1
            st.append(num)
        if done:
            break
    return answer

"""
보조 컨테이너 벨트는 앞뒤로 이동 가능, 맨 앞의 상자만 뺄 수 있음(스택)
택배상자는 순서대로 들어오고, 택배는 input order에 들어있음
순서가 맞는 택배를 순서에 맞는 택배 상자에 넣어야됨
"""