IMWeb提升营Day1 | 训练题5： 用两个栈实现队列

题目描述

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

思路

/*
 * 思路：两个栈实现一个队列
 * --入队列：直接进栈1
 * --出队列：从栈2出元素，判断队列2是否为空，如果已经弹空了点话就从栈1把元素都弹到栈2里来，然后再从栈2弹出元素
 */
class Solution
{
public:
    void push(int node) {
        stack1.push(node);
    }

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

        a = stack2.top();
        stack2.pop();
        return a;
    }

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

举例一返二：

两个队列实现一个栈的思路：
–入栈：直接进队列1
–出栈：判断队列1里是不是只有一个元素，是的话直接从队列1里面出，如果是多于一个元素的话，就把之前先进来的放到队列2里面去，把最后一个进队列1的元素弹出，实现出栈的效果，然后再把队列2的元素都放回队列1.