由两个栈组成一个队列

版权声明：转载请注明出处 https://blog.csdn.net/Zach_z/article/details/81291532

使用C++完成《程序员代码面试指南 IT名企算法与数据结构题目》

题目

编写一个类，用两个栈实现队列，支持队列操作：
- push：入队
- pop：出队（不返回值）
- front：返回最先入队的值

实现

• 栈的特点是先进后出，队列特点是先进先出

• 使用两个栈一个最为压入栈，一个作为弹出栈，压入栈只压入数据，弹出栈只弹出数据

• 数据按照1->2->3的顺序被压入进压入栈，压入栈中数据从栈顶到栈底为3 2 1

• 将压入栈数据依次弹出，并会按照顺序3->2->1进入弹出栈，此时弹出栈中数据从栈顶到栈底为1 2 3

• 数据压入顺序为1 2 3 ，弹出顺序也为1 2 3，从而实现将两个栈组成一个队列的效果

设压入栈为stackPush，输出栈为stackPop，则需要注意的是：
1. 从stackPush向stackPop中压入数据，必须一次性把stackPush中数据全部压入
2. 如果stackPop不为空，stackPush不能向stackPop中压入数据

代码实现

#include <stack>
using namespace std;

class twoStackoneQuene{
private:
    stack <int>stackPush;
    stack <int>stackPop;
public:
    twoStackoneQuene(){}
    void push(int newNum){
        stackPush.push(newNum);
    }
    void pop(){
        if(stackPush.empty()&&stackPop.empty()){
            fprintf(stderr,"The two Stacks are all empty!");
            exit(0);
        }else if(stackPop.empty()){
            while(!stackPush.empty()){
                stackPop.push(stackPush.top());
                stackPop.pop();
            }
        }
        stackPop.pop();
    }
    int front(){
        if(stackPush.empty()&&stackPop.empty()){
            fprintf(stderr,"The two Stacks are all empty!");
            exit(0);
        }else if(stackPop.empty()){
            while(!stackPush.empty()){
                stackPop.push(stackPush.top());
                stackPush.pop();
            }
        }
        return stackPop.top();
    }
};