用两个栈实现队列

用两个栈实现队列

• 参与人数：3047时间限制：1秒空间限制：32768K
• 通过比例：34.71%
• 最佳记录：0 ms|0K（来自  青哥）

题目描述

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

　　思路1（最普通的方法）：
　　1、入队时，直接压入到stack1中；
　　2、出队时，先将stack1中的所有size-1个元素压入stack2中，然后将stack1的元素弹出；最后将stack2中的元素重新压入到s1中。

　　如下图所示：

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

    int pop() {
        int t;
        while (stack1.size() > 1){
            t = stack1.top();
            stack2.push(t);
            stack1.pop();
        }
        int rt = stack1.top();
        stack1.pop();
        while(stack2.size() > 0){
            t = stack2.top();
            stack1.push(t);
            stack2.pop();
        }
        return rt;
    }

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

 

 

　　思路2（优化）：

　　1、入队时，直接压入到stack1中；
　　2、出队时，若stack2不为空，则直接弹出；若stack2为空，则将stack1的size-1个元素压入stack2中，最后弹出stack1中的栈顶元素。

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

    int pop() {
        int rt;
        if (!stack2.empty()){
            rt = stack2.top();
            stack2.pop();
        }
        else{
            while (stack1.size() > 1){
                rt = stack1.top();
                stack2.push(rt);
                stack1.pop();
            }
            rt = stack1.top();
            stack1.pop();
        }
        return rt;
    }
private:
    stack<int> stack1;
    stack<int> stack2;
};