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
반응형
'Agorithm > 백준 알고리즘' 카테고리의 다른 글
백준 알고리즘 6198번 옥상 정원 꾸미기(python) (0) | 2021.03.05 |
---|---|
백준 알고리즘 10866번 덱(python) (0) | 2021.03.05 |
백준 알고리즘 2812번 크게 만들기(python) (2) | 2021.03.05 |
백준 알고리즘 2841번 외계인의 기타 연주(python) (0) | 2021.03.05 |
백준 알고리즘 1935번 후위 표기식2(python) (0) | 2021.03.05 |