[알고리즘/백준] 10845: 큐(python)

deque를 이용해서 큐를 만들어서 풀면된다.
readline 사용해야 시간초과 안남

from collections import deque
from sys import stdin

q = deque()
for i in range(int(stdin.readline().strip())):
    a = stdin.readline().strip().split()
    if a[0] == 'push':
        q.append(a[1])

    elif a[0] == 'pop':
        print(-1 if len(q) == 0 else q.popleft())

    elif a[0] == 'size':
        print(len(q))

    elif a[0] == 'empty':
        print(1 if len(q) == 0 else 0)

    elif a[0] == 'front':
        print(-1 if len(q) == 0 else q[0])

    elif a[0] == 'back':
        print(-1 if len(q) == 0 else q[-1])

좋은 웹페이지 즐겨찾기