剑指offer题解C++【5】用两个栈实现队列

题目描述

用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型。

解题思路

队列

C++队列queue模板类的定义在< queue>头文件中，queue 模板类需要两个模板参数，一个是元素类型，一个容器类型，元素类型是必要的，容器类型是可选的，默认为deque 类型。

C++队列Queue是一种容器适配器，它给予程序员一种先进先出(FIFO)的数据结构。

C++队列Queue类成员函数如下:

• back()返回最后一个元素
• empty()如果队列空则返回真
• front()返回第一个元素
• pop()删除第一个元素
• push()在末尾加入一个元素
• size()返回队列中元素的个数

queue 的基本操作举例如下：

• queue入队，如例：q.push(x); 将x 接到队列的末端。
• queue出队，如例：q.pop(); 弹出队列的第一个元素，注意，并不会返回被弹出元素的值。
• 访问queue队首元素，如例：q.front()，即最早被压入队列的元素。
• 访问queue队尾元素，如例：q.back()，即最后被压入队列的元素。
• 判断queue队列空，如例：q.empty()，当队列空时，返回true。
• 访问队列中的元素个数，如例：q.size()

---

用两个栈实现队列的push和pop操作，注意这里给出的模板中要求pop返回int值。
入队：将元素压进栈A
出队：判断栈B是否为空，如果为空，则将栈A中所有元素pop，并push进栈B，栈B出栈；
如果不为空，栈B直接出栈。

代码

class Solution
{
public:
    void push(int node) {
        stack1.push(node);
    }

    int pop() {
        if (stack2.empty()){
            while (!stack1.empty()){
                stack2.push(stack1.top());
                stack1.pop();
            }
        }

        if (stack2.empty())
            return -1;

        int res = stack2.top();
        stack2.pop();
        return res;
    }

private:
    stack<int> stack1;
    stack<int> stack2;
};