用两个栈实现队列

最新推荐文章于 2024-10-01 22:51:02 发布

yuuunfryan

最新推荐文章于 2024-10-01 22:51:02 发布

阅读量122

点赞数 1

文章标签：数据结构

本文链接：https://blog.csdn.net/weixin_45770902/article/details/130160820

版权

该文章描述了如何使用两个栈（Stack1和Stack2）来模拟队列的行为。在Java中，push操作直接将元素压入辅助栈stack2，而pop操作时，如果主栈stack1为空，则将stack2的所有元素转移到stack1，然后从stack1弹出元素，确保了先入先出的队列特性。

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

描述

用两个栈来实现一个队列，使用n个元素来完成 n 次在队列尾部插入整数(push)和n次在队列头部删除整数(pop)的功能。队列中的元素为int类型。保证操作合法，即保证pop操作时队列内已有元素。（题目来源牛客网）

思路

两个栈，分别命名为Stack1和Stack2（辅助栈）。

队列的特性是先入先出（想象一下排队），而栈的特性是先入后出（可以想象一下弹夹），因此在完成“入队”操作时，可以将元素push进辅助站。当要进行“出队”操作时，先判断Stack1是否为空，若为空则将辅助栈的内容pop出来后push进stack1中。

import java.util.Stack;

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