[백준] 18258번 큐 2 (Python)

문제

www.acmicpc.net/problem/18258

18258번: 큐 2

 

코드

import sys
from collections import deque
N = int(input())

queue = deque()

def push(X):
    queue.append(X)

def pop():
    if not queue:
        return -1
    return queue.popleft()

def size():
    return len(queue)

def empty():
    if queue:
        return 0
    else:
        return 1

def front():
    if not queue:
        return -1
    return queue[0]

def back():
    if not queue:
        return -1
    return queue[-1]

for i in range(N):
    a=sys.stdin.readline().split()
    if a[0] == 'push':
        push(int(a[1]))
    elif a[0] == 'pop':
        print(pop())
    elif a[0] == 'size':
        print(size())
    elif a[0] == 'empty':
        print(empty())
    elif a[0] == 'front':
        print(front())
    elif a[0] == 'back':
        print(back())
    

 

코드설명

큐를 직접 구현하는 문제인데 입력을 sys로 바꿔도 시간초과가 나서 어쩔 수 없이 deque을 import해서 leftpop()을 통해 시간초과를 해결해 풀었다.