用两个栈实现队列,支持队列的基本操作。
最近想练练python
import sys
class Soul(object):
def __init__(self):
self.l1=[]
self.l2=[]
def add(self,x):
self.l1.append((x))
def peek(self):
return self.l1[0]
def poll(self):
for i in range(len(self.l1)):
if len(self.l1)-i-1>0:
self.l2.append(self.l1[len(self.l1)-i-1])
self.l1 = []
for i in range(len(self.l2)):
self.l1.append(self.l2[len(self.l2)-i-1])
self.l2 = []
res=Soul()
n=int(sys.stdin.readline().strip())
for i in range(n):
line=sys.stdin.readline().strip().split()
str_1=line[0]
if str_1=='add':
res.add(int(line[1]))
if str_1 == 'peek':
print(res.peek())
if str_1 == 'poll':
res.poll()