728x90
반응형

기본적인 큐 문제이다. sys.stdin.readline을 사용하지 않으면 시간 초과가 나온다.

import sys

class Queue():
    def __init__(self):
        self.li = []
    
    def push(self, n):
        self.li.append(n)

    def pop(self):
        print(self.li.pop(0) if self.li else -1)
    
    def size(self):
        print(len(self.li))

    def empty(self):
        print(0 if self.li else 1)
        
    def front(self):
        print(self.li[0] if self.li else -1)
        
    def back(self):
        print(self.li[-1] if self.li else -1)

queue = Queue()
for _ in range(int(sys.stdin.readline())):
    s = sys.stdin.readline().split()
    if s[0] == "push":
        queue.push(int(s[1]))
    elif s[0] == "pop":
        queue.pop()
    elif s[0] == "size":
        queue.size()
    elif s[0] == "empty":
        queue.empty()
    elif s[0] == "front":
        queue.front()
    elif s[0] == "back":
        queue.back()
728x90
반응형

+ Recent posts