剑指offer刷题记录5——用两个栈实现队列

最新推荐文章于 2024-09-27 00:45:01 发布

风影楼c

最新推荐文章于 2024-09-27 00:45:01 发布

阅读量120

点赞数

分类专栏：剑指offer 剑指offer刷题记录文章标签： java 剑指offer 栈 push pop

本文链接：https://blog.csdn.net/qq_37684824/article/details/82887283

版权

剑指offer 同时被 2 个专栏收录

32 篇文章 0 订阅

订阅专栏

剑指offer刷题记录

32 篇文章 0 订阅

订阅专栏

本文记录了使用两个栈实现队列的Push和Pop操作的方法，包括两种解法。解法一在元素进出队列时涉及较多转移操作，而解法二更为简洁，直接利用1号栈存入元素，2号栈取出元素，确保了FIFO特性。优化后的解法在运行时间和内存占用上都有所提升。

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

题目描述：

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

解法一：

import java.util.Stack;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    public void push(int node) {
        while(!stack2.isEmpty()) {
            stack1.push(stack2.pop());
        }
        stack1.push(node);
    }
    
    public int pop() {
        while(!stack1.isEmpty()) {
            stack2.push(stack1.pop());
        }
        return stack2.pop();
    }
}

每次来一个元素需要进队列时候，1号栈先把所有的2号栈的元素取过来，再把待进栈元素存入1号栈。

出队列时，将所有1号栈的元素娶过来放到2号栈，然后再把2号栈元素取出一个来。

在看了其他解法之后发现这种解法较为繁琐，因为其核心就是1号栈存元素，2号栈取元素。所以产生解法二。

运行时间：18ms（不稳定）

占用内存：9028k（不稳定）

解法二：

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() && stack2.isEmpty()){
            throw new RuntimeException("Queue is empty!");
        }
        if(stack2.isEmpty()){
            while(!stack1.isEmpty()){
                stack2.push(stack1.pop());
            }
        }
        return stack2.pop();
    }
}

解法二相较解法一来说，少了“将所有元素放入一个栈，再进行新元素操作”的过程。

存元素直接存在1号栈中

取元素从2号栈中取，如果2号栈没有元素，将1号栈的元素全部转移到2号栈

因为每个元素在被取之前，其与其之前被取得元素都会经历一次从1号栈到2号栈的转移，因此符合FIFO

要注意考虑调用pop()方法时，两个栈中皆无元素的情况

运行时间：9ms

占用内存：9464k