剑指offer——用两个栈实现队列

1. 题目描述

  用来两个栈来实现一个队列，并完成队列的push和pop操作。队列中的元素全为整数。

2. 题目分析

  首先要清晰两个概念：栈和队列。栈是先入后出，队列是先入先出。所以，实现队列的push操作较为简单，即向一个栈中依次push数据即可。而队列的pop操作，需要弹出最早进入队列的元素，即最早被压入栈stack1的元素。可见，该元素在stack1的最低端，而栈是先入后出，所以若想弹出这个元素，需要将该栈stack1整体反转过来。最简单的反转方法，即依次将元素出栈，并压入到一个新栈中。此时stack1的最底元素即为stack2的最顶元素，直接在stack2中弹出即可。弹出该元素后，在依次将元素弹出并压回stack1，即完成了此次pop操作。

3. 题目解析——cpp

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

    int pop() {
        while(!stack1.empty()) {
            stack2.push(stack1.top());
            stack1.pop();
        }
        int ans = stack2.top();
        stack2.pop();
        while(!stack2.empty()) {
            stack1.push(stack2.top());
            stack2.pop();
        }
        return ans;
    }

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