[프로그래머스] Lv.2 택배상자 [Python, 파이썬]

• 문제 : https://school.programmers.co.kr/learn/courses/30/lessons/131704
• 🔑 스택
  • 1번부터 나열된 박스(i)를 스택에 담는다.
  • 스택의 맨 위 박스가 택배 기사가 원하는 박스면 빼서 트럭에 담는다.
  • 원하는 박스가 아니면 다음 박스 담으러 이동

 

def solution(order):
    answer = 0
    stack = []  # 보조 컨베이어 벨트
    i = 1

    while i < len(order)+1:
        stack.append(i)                     # 보조 컨베이어 벨트에 i번 박스 담음
        while stack[-1] == order[answer]:   # 보조 컨베이어 벨트 맨 위 박스가 택배 기사가 원하는 박스면
            answer += 1                     # 트럭에 담음
            stack.pop()                     # 보조 컨베이어 벨트에서 제거
            if not stack:
                break
        i += 1
        
    return answer