LeetCode255.用队列实现栈

题目传送门:Leetcode255.用队列实现栈

请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通栈的全部四种操作(pushtoppopempty)。

实现 MyStack 类:

  • void push(int x) 将元素 x 压入栈顶。
  • int pop() 移除并返回栈顶元素。
  • int top() 返回栈顶元素。
  • boolean empty() 如果栈是空的,返回 true ;否则,返回 false

注意:

  • 你只能使用队列的基本操作 ------ 也就是 push to backpeek/pop from frontsizeis empty 这些操作。
  • 你所使用的语言也许不支持队列。 你可以使用 list (列表)或者 deque(双端队列)来模拟一个队列 , 只要是标准的队列操作即可。

示例:

复制代码
输入:
["MyStack", "push", "push", "top", "pop", "empty"]
[[], [1], [2], [], [], []]
输出:
[null, null, null, 2, 2, false]

解释:
MyStack myStack = new MyStack();
myStack.push(1);
myStack.push(2);
myStack.top(); // 返回 2
myStack.pop(); // 返回 2
myStack.empty(); // 返回 False

提示:

  • 1 <= x <= 9
  • 最多调用100pushpoptopempty
  • 每次调用 poptop 都保证栈不为空

**进阶:**你能否仅用一个队列来实现栈。

试题解析:

已知队列是先进先出,栈是先进后出。

当我们寻找栈顶元素时,实际上是要将当前队列的尾元素输出,但队列的pop()函数只能弹出队首,这时便可以使用第二个辅助队列。
具体方案:

定义两个队列q1,q2,q1为存放数据的队列,q2是辅助队列,每一步操作之后都要将数据存回q1

进行push操作时,在q1中插入元素

进行pop操作时:

1、将q1中的除了队尾之外的元素,全部插入到q2队列中

2、在q1中删除剩下的元素,即队尾元素

3、将q2队列中的元素再插回到q1中

cpp 复制代码
class MyStack {
public:
    queue<int> q1;
    queue<int> q2;
    MyStack() {
    }
    
    void push(int x) {
        q1.push(x);
    }
    
    int pop() {
        int n = 0;
        while(n < q1.size() - 1){//循环到q1只剩一个元素
            q2.push(q1.front());
            q1.pop();
        }
        int num = q1.front();
        q1.pop();
        //将数据存回q1
        while(!q2.empty()){
            q1.push(q2.front());
            q2.pop();
        }
        return num;
    }
    
    int top() {
        return q1.back();
    }
    
    bool empty() {
        if(q1.empty()){
            return true;
        }
        return false;
    }
};
更好方案

从以上方法我们可以观察到,q1是存放数据的队列,q2为辅助队列,每一次执行删除之后都要将q2的数据存回q1,接下来的push,pop操作都是从q1开始

那么我们可不可以在每一次pop中都少一次存回q1的操作,而将之后的push,pop操作开始于q2呢?

已知我们每一次转移元素操作后,都会有一个队列为空,那么pop操作时,我们只需要从不为空的队列开始操作即可

至于push操作,在最开始时,q1,q2都为空时,我们将元素添加到q1,对于之后的操作,我们还是只需要从不为空的队列开始插入元素即可。

cpp 复制代码
class MyStack {
public:
    queue<int> q1;
    queue<int> q2;
    MyStack() {}
    
    void push(int x) {
        //若q1,q2都不为空,则插入到q1后
        if(q1.empty()&&q2.empty()){
            q1.push(x);
        }
        else{
            //选择不空的队列插入元素
            if(!q1.empty()) q1.push(x);
            else if(!q2.empty()) q2.push(x);
        }
        
    }
    
    int pop() {
        int n = 0;
        int num;
        //选择不空的队列操作
        if(!q1.empty()){
            while(n < q1.size() - 1){
                q2.push(q1.front());
                q1.pop();
            }
            num = q1.front();
            q1.pop();
        }
        else if(!q2.empty()){
            while(n < q2.size() - 1){
                q1.push(q2.front());
                q2.pop();
            }
            num = q2.front();
            q2.pop();
        }
        return num;
    }
    
    int top() {
        //选择不空的队列操作
        if(q1.empty()){
            while(!q2.empty()){
                q1.push(q2.front());
                q2.pop();
            }
            return q1.back();
        }
        else if(q2.empty()){
            while(!q1.empty()){
                q2.push(q1.front());
                q1.pop();
            }
            return q2.back();
        }
        return 0;
    }
    
    bool empty() {
        if(q1.empty()&&q2.empty()){
            return true;
        }
        return false;
    }
};
相关推荐
foundbug99910 分钟前
基于混合整数规划的电池容量优化 - MATLAB实现
数据结构·算法·matlab
unicrom_深圳市由你创科技29 分钟前
上位机开发常用的语言 / 框架有哪些?
c++·python·c#
自我意识的多元宇宙43 分钟前
树、森林——树与二叉树的应用(哈夫曼树的构造)
数据结构
|_⊙1 小时前
C++ 智能指针
开发语言·c++
memcpy01 小时前
LeetCode 2452. 距离字典两次编辑以内的单词【暴力;字典树】中等
算法·leetcode·职场和发展
水蓝烟雨1 小时前
2071. 你可以安排的最多任务数目
数据结构·链表
Jasmine_llq2 小时前
《B4356 [GESP202506 二级] 数三角形》
开发语言·c++·双重循环枚举算法·顺序输入输出算法·去重枚举算法·整除判断算法·计数统计算法
山栀shanzhi2 小时前
在做直播时,I帧的间隔(GOP)一般是多少?
网络·c++·面试·ffmpeg
王老师青少年编程2 小时前
csp信奥赛C++高频考点专项训练之贪心算法 --【排序贪心】:魔法
c++·算法·贪心·csp·信奥赛·排序贪心·魔法
wearegogog1232 小时前
基于和差波束法的单脉冲测角MATLAB实现
人工智能·算法·matlab