[파이썬] 프로그래머스 : 택배 상자 (Lv.2)

[파이썬] 프로그래머스 : 택배 상자 (Lv.2)

 

프로그래머스

 

 

 

---

풀이

방향성 생각

큐, 스택문제. 들어오는 입력에 대해서 order와 비교해주면서 풀이

answer에는 넣을 수 있는 택배, stack에는 현재 order와는 달라서 잠시 넣을 택배, income은 새로 들어오는 택배.

 

전체코드

from collections import deque

def solution(orders):
    answer,stack = [],[]
    orders = deque(orders)
    income = list(range(1,len(orders)+1))

    order = orders.popleft() # 새로운 주문
    for box in income:
        if box == order: # 새로운 주문이 들어온 박스와 같으면
            answer.append(box) # 정답에 추가
            if orders: # 주문이 남았으면
                order = orders.popleft() # 새로운 주문으로 변경
            while stack: # 스택이 있을 때
                if stack[-1] == order: # 주문이 스택의 마지막과 같으면
                    answer.append(stack.pop()) # 스택 맨 뒤를 답에
                    if orders: # 주문이 남았으면
                        order = orders.popleft() # 새로운 주문으로 변경
                else: # 스택이 주문과 다르면 탈출
                    break
        else: # 새로운 주문이 박스와 다르면 스택으로
            stack.append(box)

    return len(answer)

 

코멘트