[백준 10845] 파이썬

https://www.acmicpc.net/problem/10845

10845번: 큐

import sys
from collections import deque #덱 사용
input=sys.stdin.readline 
n=int(input())
queue=deque() 

for _ in range(n):
    cmd,*val=input().split() # cmd: 명령받기, val: 숫자받기(숫자는 없을수도 있으므로 가변인자로)
    if cmd=='push':
        queue.append(int(val[0]))
    elif cmd=='pop':
        if queue:
            print(queue.popleft())
        else:
            print(-1)
    elif cmd=='size':
        print(len(queue))
    elif cmd=='empty':
        if queue:
            print(0)
        else:
            print(1)
    elif cmd=='front':
        if queue:
            print(queue[0])
        else:
            print(-1)
    elif cmd=='back':
        if queue:
            print(queue[-1])
        else:
            print(-1)

정리

-양쪽에서 값을 넣고 빼기위해 deque 라이브러리 사용. (큐, 스택 커버 가능)

-deque을 사용하면 popleft(),popright(),append()등 대부분 연산을 O(1) 에 가능하다.

-input을 sys.stdin.readline으로 변경하지 않으면 시간초과 발생.