BM42 用两个栈实现队列

小火柴66

已于 2022-03-28 23:51:56 修改

阅读量157

点赞数

分类专栏：算法 # 堆/栈/队列文章标签： leetcode 算法 java 数据结构

于 2022-03-28 23:02:06 首次发布

本文链接：https://blog.csdn.net/liulong999/article/details/123808909

版权

算法同时被 2 个专栏收录

53 篇文章 0 订阅

订阅专栏

堆/栈/队列

7 篇文章 0 订阅

订阅专栏

在这里插入图片描述
第一种：压入的时候只压stack1栈，弹出的时候只弹stack2栈，弹出的时候只要stack2为null就从stack1里压到stack2里(注意一次性压完)，stack2不为null就不从stack1里压。时间复杂度O(1)，空间复杂度O(n)。

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 (stack2.isEmpty()) {
            while (!stack1.isEmpty()) {
                stack2.push(stack1.pop());
            }
        }
        return stack2.pop();
    }
}

第二种：维护两个栈，每次都是从stack1压入，从stack2弹出

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();
    }
}