算法实现之用两个栈实现队列

最新推荐文章于 2024-01-20 20:46:33 发布

原创最新推荐文章于 2024-01-20 20:46:33 发布 · 256 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Java #leetcode

Java 同时被 3 个专栏收录

67 篇文章

订阅专栏

算法与数据结构

42 篇文章

订阅专栏

LeetCode小记

33 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现队列的方法，通过将元素压入stack1，在需要取出数据时，若stack2为空，则将stack1中的数据全部转移至stack2，从而保证了队列的先进先出特性。

题目

用两个栈来实现一个队列，完成队列的push和pop操作。

思路

队列的性质就是先进先出，那么可以把所有元素压如stack1中，取数据的时候，如果stack2为空，就把stack中的数据取出压入stack2中，这样最早加入stack1中的数据就在stack2的最上面，直接pop即可。

实现

public class Solution {

    Stack<Object> stack1 = new Stack<>();
    Stack<Object> stack2 = new Stack<>();

    public static void main(String[] args) {
        Solution solution = new Solution();
        for (int i = 0; i < 8; i++) {
            solution.push(i);
        }
        for (int i = 0; i < 8; i++) {
            System.out.println(solution.pop());
        }
    }

    public void push(Object value) {
        stack1.push(value);
    }

    public Object pop() {
        if (stack2.isEmpty()) {
            while (!stack1.isEmpty()) {
                stack2.push(stack1.pop());
            }
        }
        return stack2.pop();
    }

}