两个栈实现队列

题目

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作（push、pop、peek、empty）：
实现 MyQueue 类：
void push(int x) 将元素 x 推到队列的末尾
int pop() 从队列的开头移除并返回元素
int peek() 返回队列开头的元素
boolean empty() 如果队列为空，返回 true ；否则，返回 false

说明：
你只能使用标准的栈操作 —— 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
你所使用的语言也许不支持栈。你可以使用 list 或者 deque（双端队列）来模拟一个栈，只要是标准的栈操作即可。

提示：
1 <= x <= 9
最多调用 100 次 push、pop、peek 和 empty
假设所有操作都是有效的 （例如，一个空的队列不会调用 pop 或者 peek 操作）

示例
输入：
[“MyQueue”, “push”, “push”, “peek”, “pop”, “empty”]
[[], [1], [2], [], [], []]
输出：
[null, null, null, 1, 1, false]

解释： MyQueue myQueue = new MyQueue(); myQueue.push(1); // queue is: [1]
myQueue.push(2); // queue is: [1, 2] (leftmost is front of the queue)
myQueue.peek(); // return 1 myQueue.pop(); // return 1, queue is [2]
myQueue.empty(); // return false

基本概念

栈对数据的存储顺序：先入后出（FILO）。最先进入的元素被放在栈底，最后进入的元素被放在栈顶。

队列对数据的存储顺序：先入先出（FIFO）。数据是从队列的尾部插入，从队列的头部取出。

算法的思想（pop函数的实现）

一个栈（记栈A）用于存储插入的数据，一个栈（记为栈B）用于弹出数据
针对取数据，需要分四种情况进行讨论：

1.栈A和栈B都有数据（数字表示数据的插入顺序，小数字表示先插入）

如果栈B有数据，则不能将栈A的数据放进栈B，因为现在栈B的数据是比栈A的数据先插入到“队列“中的。应该先将栈B的数据取完，再将栈A的数据压入栈B中。

2.栈B为空，栈A有数据

这种情况就需要将栈A的数据全部压入栈B，然后取出栈B的顶部

3.栈A无数据，栈B有数据

这种情况直接取出栈B的顶部数据即可

4.栈A和栈B都无数据

这种情况说明队列无数据，返回错误即可。

代码实现

//用C++实现
class MyQueue {
public:
    stack<int> stack1, stack2;
    MyQueue() {
    }
    
    void push(int x) {
        stack1.push(x);
    }
    
    int pop() {
        if(!stack2.empty()){    //判断第二个栈是不是空的，如果不是，就从取出该栈的顶部
            int temp = stack2.top();
            stack2.pop();
            return temp;
        }
        else{   //第二个栈是空的，那么就将第一个栈的元素依次放入第二个栈，然后取出第二个栈的顶部
            if(!stack1.empty()){
                int temp;
                while(!stack1.empty()){
                    temp = stack1.top();
                    stack1.pop();
                    stack2.push(temp);
                }
                stack2.pop();
                return temp;
            }
        }
        return 0;
    }
    
    int peek() {
        if(!stack2.empty()){    //判断第二个栈是不是空的，如果不是，就从读出该栈的顶部
            int temp = stack2.top();
            return temp;
        }
        else{   //第二个栈是空的，那么就将第一个栈的元素依次放入第二个栈，然后读出第二个栈的顶部
            if(!stack1.empty()){
                int temp;
                while(!stack1.empty()){
                    temp = stack1.top();
                    stack1.pop();
                    stack2.push(temp);
                }
                return temp;
            }
        }
        return 0;
    }
    
    bool empty() {      //两个栈只要有其中一个是非空的，就说明队列肯定是非空的
        if(!stack2.empty()){    //判断第二个栈是不是空的
            return false;
        }
        else{   //第二个栈是空的，判断第一个栈
            if(!stack1.empty()){
                return false;
            }
            else{
                return true;
            }
        }
    }
};

运行结果