[백준][Python] 10845번 큐

10845번: 큐

 

코드

import sys
from collections import deque
n = int(input())
queue = deque([])

for i in range(n):
    com =  sys.stdin.readline().split()
    if com[0] == 'push':
        queue.append(com[1])
    elif com[0] == 'pop':
        if len(queue) == 0:
            print(-1)
        else:
            print(queue.popleft())
    elif com[0] == 'size':
        print(len(queue))
    elif com[0] == 'empty':
        if len(queue) == 0:
            print(1)
        else:
            print(0)
    elif com[0] == 'front':
        if len(queue) == 0:
            print(-1)
        else:
            print(queue[0])
    elif com[0] == 'back':
        if len(queue) == 0:
            print(-1)
        else:
            print(queue[-1])

 

설명

deque를 사용해야 pop을 할 때 시간초과가 발생하지 않는다.