堆栈模拟队列(python)

设已知有两个堆栈S1和S2,请用这两个堆栈模拟出一个队列Q。

所谓用堆栈模拟队列,实际上就是通过调用堆栈的下列操作函数:

int IsFull(Stack S):判断堆栈S是否已满,返回1或0;
int IsEmpty (Stack S ):判断堆栈S是否为空,返回1或0;
void Push(Stack S, ElementType item ):将元素item压入堆栈S;
ElementType Pop(Stack S ):删除并返回S的栈顶元素。
实现队列的操作,即入队void AddQ(ElementType item)和出队ElementType DeleteQ()。

输入格式:
输入首先给出两个正整数N1和N2,表示堆栈S1和S2的最大容量。随后给出一系列的队列操作:A item表示将item入列(这里假设item为整型数字);D表示出队操作;T表示输入结束。

输出格式:
对输入中的每个D操作,输出相应出队的数字,或者错误信息ERROR:Empty。如果入队操作无法执行,也需要输出ERROR:Full。每个输出占1行。

输入样例:
3 2
A 1 A 2 A 3 A 4 A 5 D A 6 D A 7 D A 8 D D D D T
结尾无空行
输出样例:
ERROR:Full
1
ERROR:Full
2
3
4
7
8
ERROR:Empty
结尾无空行

首先这题需要搞懂题目的意思
两个栈,一个大的一个小的,就以2和3为例
我们先往小的栈里面添加元素直到栈满,然后把小栈内的元素都放入大栈中,这个时候如果小栈又满了,就不能往大栈中继续放了,因为队列只能一头进一头出,这个时候就输出满,这就解释了为什么例题中加了四个就满了。删除的话删除大栈里面的元素,当删光了之后,把小栈里面的元素加到大栈中,如果大栈和小栈都没元素,再删就输出空。

代码如下:

m,n=input().split()
m=min(int(m),int(n))
lst=input().split()
lst1,lst2=[],[]
for i in range(len(lst)-1):
    if lst[i]=='A':
        if len(lst1)==m and lst2==[]:
            lst1.reverse()
            lst2=lst1[:]
            lst1.clear()
            lst1.append(lst[i+1])
        elif len(lst1)==m and lst2!=[]:
            print('ERROR:Full')
        elif len(lst1)!=m:
            lst1.append(lst[i+1])
    elif lst[i].isdigit():
        continue
    else:
        if lst2!=[]:
            print(lst2.pop())
        elif lst2==[] and lst1!=[]:
            lst1.reverse()
            lst2=lst1[:]
            lst1.clear()
            print(lst2.pop())
        elif lst1==[] and lst2==[]:
            print('ERROR:Empty')
上一篇:二十一天好习惯 第一期-6


下一篇:LeetCode题解(1650):二叉树的最近公共祖先III(Python)