堆栈模拟队列

7-22 堆栈模拟队列 （25 分）

设已知有两个堆栈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

 

 

 

 

#include <iostream>
#include <stack>
using namespace std;
class myQueue{
private:
    int Max1, Max2;
    bool flag;
    stack<int> st1;
    stack<int> st2;
public:
    myQueue(int max1, int max2){
        flag = max1 > max2 ? true : false; 
        Max1 = max1;
        Max2 = max2;
    }
    void Push(int X){
        if(flag){
            //此时栈1为出栈，栈2为入栈 
            if(st2.size() == Max2 && !st1.empty()){//栈2满，栈1不为空 
                cout << "ERROR:Full" << endl;
            }else if(st2.size() == Max2){//栈2满，栈1空 
                while(!st2.empty()){//将栈2全部导入栈1 
                    st1.push(st2.top());
                    st2.pop();
                }
                st2.push(X);//将X加入栈2    
            }else{//不满足上面俩种情况，直接入栈1 
                st2.push(X);
            }
        }else{
            //此时栈2为出栈，栈1为入栈 
            if(st1.size() == Max1 && !st2.empty()){//栈1满，栈2不为空 
                cout << "ERROR:Full" << endl;
            }else if(st1.size() == Max1){//栈1满，栈2空 
                while(!st1.empty()){//将栈1全部导入栈2 
                    st2.push(st1.top());
                    st1.pop();
                }
                st1.push(X);//此时可以入栈1    
            }else{
                st1.push(X);//直接入栈1 
            }
        }
    }
    void Pop(){ 
        if(flag){
            //栈1为出栈，栈2为入栈 
            if(st1.empty() && st2.empty()){//栈1栈2都为空 
                cout << "ERROR:Empty";
            
            }else if(st1.empty()){//栈1为空，栈2不为空 
                    while(!st2.empty()){//将栈2中的元素直接导入栈1 
                        st1.push(st2.top());
                        st2.pop();
                    }
                    cout << st1.top();//弹出栈1中的栈顶元素 
                    st1.pop();
            }else{
                cout << st1.top();//不是上面俩种情况就直接弹出栈1 
                st1.pop();
            }
        }else{//栈2为出栈，栈1为入栈 
            if(st1.empty() && st2.empty()){
                cout << "ERROR:Empty";
            
            }else if(st2.empty()){
                    while(!st1.empty()){//将栈1元素全部导入栈2 
                        st2.push(st1.top());
                        st1.pop();
                    }
                    cout << st2.top();//弹出栈2栈顶元素 
                    st2.pop();
            }else{
                cout << st2.top();//不是上面两种情况直接弹出栈1栈顶元素 
                st2.pop();
            }
        }    
    }    
};
int main(){
    int N1, N2;
    cin >> N1 >> N2;//输入两个栈的容量 
    myQueue q(N1, N2);//构造用栈构造的队列 
    char Op;
    do{
        
        int X;
        cin >> Op;//读入操作 

        switch(Op){
            case 'A': cin >> X; q.Push(X);
            break;
            case 'D': q.Pop(); cout << endl;
        } 
    }while(Op != 'T');
    return 0;
}