两个栈实现一个队列

本题先前在复习王道时碰到过,当时因为写伪代码,只是想让两个栈来回倒,没有考虑其他问题;这次在做题时直接写系统提示我栈调用次数太多,段错误了。错误代码如下:

class Solution
{
public:
    void push(int node) {
        stack1.push(node);
    }
 
     int pop() {
        int head;
        while(stack1.empty()){
            stack2.push(stack1.top());
            stack1.pop();
        }
        head = stack2.top;
        stack2.pop();
        while(stack2.empty()){
            stack1.push(stack2.top());
            stack2.pop();
        }
        return head;
    }
    }
 
private:
    stack<int> stack1;
    stack<int> stack2;
};

 在参考别人处理该问题的方法后,我发现我没思考到的问题在于:

1.stack2如果非空则其pop出的数字序列仍为相应队列pop操作应有的部分序列,无需每pop一次就重置一次stack2

2.在stack2空,stack1往stack2输送数字时,无需再将数值保存回stack1,stack2就是队列序列的储存体,stack1只是中转站。

下面为正确代码:

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

    int pop() {
        int head;
        if (stack1.empty() && stack2.empty()) return -1;
        //以s1和s2均空表示队列空
        if (!stack2.empty()){
            head = stack2.top();
            stack2.pop();
        }
        //s2非空则从s2中pop
        else{
            while (stack1.size() > 0){
                stack2.push(stack1.top());
                stack1.pop();
            }
            head = stack2.top();
            stack2.pop();
        }
        //s2空,s1非空则说明上一段队列序列已清空,需要重新往s2输送队列序列
        //注意,伴随着s2数字的装填,中转站s1也要逐步清空
        return head;
    }

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

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值