剑指offer之两个栈模拟队列操作

最新推荐文章于 2022-08-05 19:52:29 发布

havedream_one

最新推荐文章于 2022-08-05 19:52:29 发布

阅读量559

点赞数

CC 4.0 BY-SA版权

分类专栏：面试剑指offer 文章标签：栈队列剑指offer

本文链接：https://blog.youkuaiyun.com/havedream_one/article/details/48206881

面试同时被 2 个专栏收录

34 篇文章

订阅专栏

剑指offer

5 篇文章

订阅专栏

本文探讨了如何使用两个栈实现队列的功能，并通过改进代码减少了执行'出队'操作时的冗余步骤。具体而言，文章解释了原始实现中存在的问题，并提出了一种优化方案来提升效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

队列的特点是先进先出，而栈的特点是先进后出
所以当用两个栈模拟队列的时候，为了依然保持队列的特点，可以将一个栈作为主栈，一个栈作为备用栈

当执行“入队”的时候，可以直接进入主栈；
当执行“出对”的时候，可以先将主栈元素依次push进备用栈直到主栈剩余一个元素（即是要出队的元素）

import java.util.Stack;

public class Solution {
    //主栈
    Stack<Integer> stack1 = new Stack<Integer>();
    //备用栈
    Stack<Integer> stack2 = new Stack<Integer>();

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

    public int pop() {
        if(stack1.isEmpty()){
            return 0;
        }
        while(stack1.size() > 1 ){
            stack2.push(stack1.pop());
        }
        int tmp = stack1.pop();
        while(!stack2.isEmpty()){
            stack1.push(stack2.pop());
        }
        return tmp;
    }
}

但是还是有问题的？问题在于执行“出队”时，都要先腾挪到备用栈，执行完再腾挪回主栈。那么是不是多此一举呢？显然，我们可以发现当腾挪到备用栈之后，备用栈的元素顺序就是出队顺序。
所以需要改进代码：

public class Solution {
    //入队栈
    Stack<Integer> stack1 = new Stack<Integer>();
    //出队栈
    Stack<Integer> stack2 = new Stack<Integer>();

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

    public int pop() {
        //出队栈空，需要将入队栈腾挪到出队栈
        if(stack2.isEmpty()){
            if(stack1.isEmpty()){
                return 0;
            }
            while(!stack1.isEmpty()){
                stack2.push(stack1.pop());   
            } 
        }
        return stack2.pop();
    }
}