用两个栈实现队列（优化方法）

最新推荐文章于 2024-04-22 23:20:09 发布

流星空

最新推荐文章于 2024-04-22 23:20:09 发布

阅读量232

点赞数

分类专栏：剑指offer 文章标签：栈 leetcode stack

本文链接：https://blog.csdn.net/m0_46441208/article/details/118416099

版权

剑指offer 专栏收录该内容

57 篇文章 1 订阅

订阅专栏

本文探讨了一种CQueue数据结构的实现，通过双栈优化提高appendTail和deleteHead操作的效率。原始实现中，每次删除头元素时会将整个栈转移，导致效率低下。改进方案是保持一个栈用于插入，另一个栈用于删除，当删除栈为空时，才将插入栈的元素转移到删除栈。实验表明，这种优化显著提升了操作效率。

摘要由CSDN通过智能技术生成

题目：

剑指offer09

实现思路：

一个栈作为中间栈，另一个栈作为主栈。

class CQueue {
public:
    stack<int>a,b;
    
    CQueue() {

    }
    
    void appendTail(int value) {
        a.push(value);
    }
    
    int deleteHead() {
        if(!a.empty()){
            while(!a.empty()){
                b.push(a.top());
                a.pop();
            }
            int c=b.top();
            b.pop();
            while(!b.empty()){
                a.push(b.top());
                b.pop();
            }
            return c;
        }
        return -1;
    }
};

结果：

思路1结果
可见，效率太低！

看了评论区老哥们的思路，有一个优化思路吸引了我的注意：

in栈只负责进，out栈只负责出；
当out栈为空时：
in栈也为空时，输出-1；
int栈不为空时，in栈全导入out栈；
out栈不为空时，弹出out栈顶部元素。

class CQueue {
public:
    stack<int>in,out;
    
    CQueue() {

    }
    
    void appendTail(int value) {
        in.push(value);
    }
    
    int deleteHead() {
        if(out.empty()){
            if(in.empty()){
                return -1;
            }
            while(!in.empty()){
                out.push(in.top());
                in.pop();
            }
        }
        int t=out.top();
        out.pop();
        return t;
    }
};