用两个栈实现队列——NC.76

最新推荐文章于 2024-02-26 09:50:22 发布

原创最新推荐文章于 2024-02-26 09:50:22 发布 · 238 阅读

0 ·

CC 4.0 BY-SA版权

牛客题霸算法篇专栏收录该内容

36 篇文章

订阅专栏

这个Java程序展示了如何利用两个栈作为数据结构来模拟队列的Push和Pop操作。当队列需要出栈（Pop）时，如果第二个栈不为空则直接从栈顶弹出元素，否则将第一个栈的所有元素依次弹出并压入第二个栈，然后从第二个栈弹出顶部元素，实现了队列的先进先出原则。

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

题目描述

用两个栈来实现一个队列，完成队列的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) {
        stack1.push(node);
    }
    
    public int pop() {
        if (!stack2.isEmpty()) {    // 当stack2不为空的时候，直接出栈
            return stack2.pop();
        } else {    // 当stack2为空的时候，就把stack1里面的数据pop出来存在stack2中
            while (!stack1.isEmpty()) {
                int node = stack1.pop();
                stack2.push(node);
            }
            return stack2.pop();
        }
    }
}