用两个栈实现一个队列

最初重温了下数据结构的栈和队列的基本使用,想要在LeetCode上刷下题巩固,首先选了道标注简单难度的题,结果最开始就看懵了,不太理解它的输入输出是怎么理解,下面我们就来看一下题目描述:
在这里插入图片描述
以下是初始代码模板:
在这里插入图片描述
下面我来带没看懂的朋友理解一下这个输入输出的意思:
输入: [“CQueue”,“appendTail”,“deleteHead”,“deleteHead”] 这里是要执行的方法,从左到右执行

[[],[3],[],[]]对应上面的方法,是上面方法的参数。CQueue和deleteHead方法不需要指定数字,只有添加才需要指定数字

1.创建队列,返回值为null

2.将3压入栈,返回值为null

3.将栈底的元素删除,也就是消息队列中先进来的元素,所以是deleteHead,返回该元素的数值,所以为3

4.继续删除栈底的元素,但是没有元素了,所以返回-1

所以就有了下面的输出 输出:[null,null,3,-1]

示例 2: 输入: [“CQueue”,“deleteHead”,“appendTail”,“appendTail”,“deleteHead”,“deleteHead”]

[[],[],[5],[2],[],[]]

1.创建队列,返回值为null

2.删除栈底的元素,但是没有元素,所以返回-1

3.把5压入栈,返回null

4.把2压入栈,返回null

5.删除栈底的一个元素,也就是消息队列中先进来的元素,所以是deleteHead,就是最先进来的5,返回值为5,

6.删除栈底的一个元素,就是后进来的2,返回值为2,

所以就有了下面的输出

输出:[null,-1,null,null,5,2]

有没有发现先进来的数字,首先显示出来了,但是题目中说要使用栈,栈是先进后出的,使用栈来实现先进先出,在这里使用两个栈就好了,从一个进来再到另一个栈,这样顺序就是先进先出了。

这里给大家一个思路,即如何用两个栈去实现一个队列:

维护两个栈,第一个栈支持插入操作,第二个栈支持删除操作。

根据栈先进后出的特性,我们每次往第一个栈里插入元素后,第一个栈的底部元素是最后插入的元素,第一个栈的顶部元素是下一个待删除的元素。为了维护队列先进先出的特性,我们引入第二个栈,用第二个栈维护待删除的元素,在执行删除操作的时候我们首先看下第二个栈是否为空。如果为空,我们将第一个栈里的元素一个个弹出插入到第二个栈里,这样第二个栈里元素的顺序就是待删除的元素的顺序,要执行删除操作的时候我们直接弹出第二个栈的元素返回即可。

成员变量
维护两个栈 stack1 和 stack2,其中 stack1 支持插入操作,stack2 支持删除操作

构造方法
初始化 stack1 和 stack2 为空

插入元素
插入元素对应方法 appendTail
stack1 直接插入元素

删除元素
删除元素对应方法 deleteHead
如果 stack2 为空,则将 stack1 里的所有元素弹出插入到 stack2 里
如果 stack2 仍为空,则返回 -1,否则从 stack2 弹出一个元素并返回

在实现中我们可以用java.util的Deque接口或者自己去实现栈,在这里我们用Deque来实现,自己实现栈的话可以去看我前面栈结构的实现部分。

这里简要介绍一下java自带的结构:
Deque与支持从数据结构的任一端添加或删除元素的双端队列相关,它可以用作队列(先进先出/ FIFO)或堆栈(LIFO)。这些比Stack和LinkedList更快。
这是Java中Deque接口的层次结构:
在这里插入图片描述
Deque的方法
add(element):向尾部添加元素。
addFirst(element):向头部添加元素。
addLast(element):向尾部添加元素。
offer(element):向尾部添加一个元素并返回一个布尔值来解释插入是否成功。
offerFirst(element):向头部添加一个元素并返回一个布尔值来解释插入是否成功。
offerLast(element):向尾部添加一个元素并返回一个布尔值来解释插入是否成功。
iterator():返回此双端队列的迭代器。
descendingIterator():返回一个迭代器,该迭代器具有此双端队列的相反顺序。
push(element):向头部添加元素。
pop(element):从头部移除一个元素并返回它。
removeFirst():删除头部的元素。
removeLast():删除尾部的元素。
poll():检索并删除此双端队列表示的队列的头部(换句话说,此双端队列的第一个元素),如果此双端队列为空,则返回null。
pollFirst():检索并删除此双端队列的第一个元素,如果此双端队列为空,则返回null。
pollLast():检索并删除此双端队列的最后一个元素,如果此双端队列为空,则返回null。
peek():检索但不删除此双端队列表示的队列的头部(换句话说,此双端队列的第一个元素),如果此双端队列为空,则返回null。
peekFirst():检索但不删除此双端队列的第一个元素,如果此双端队列为空,则返回null。
peekLast():检索但不删除此双端队列的最后一个元素,如果此双端队列为空,则返回null。

java实现如下:

class CQueue {
    Deque<Integer> stack1;
    Deque<Integer> stack2;
    
    public CQueue() {
        stack1 = new LinkedList<Integer>();
        stack2 = new LinkedList<Integer>();
    }
    
    public void appendTail(int value) {
        stack1.push(value);
    }
    
    public int deleteHead() {
        // 如果第二个栈为空
        if (stack2.isEmpty()) {
            while (!stack1.isEmpty()) {
                stack2.push(stack1.pop());
            }
        } 
        if (stack2.isEmpty()) {
            return -1;
        } else {
            int deleteItem = stack2.pop();
            return deleteItem;
        }
    }
}

/**
 * Your CQueue object will be instantiated and called as such:
 * CQueue obj = new CQueue();
 * obj.appendTail(value);
 * int param_2 = obj.deleteHead();
 */

复杂度分析

时间复杂度:对于插入和删除操作,时间复杂度均为 O(1)O(1)。插入不多说,对于删除操作,虽然看起来是 O(n)O(n) 的时间复杂度,但是仔细考虑下每个元素只会「至多被插入和弹出 stack2 一次」,因此均摊下来每个元素被删除的时间复杂度仍为 O(1)O(1)。

空间复杂度:O(n)O(n)。需要使用两个栈存储已有的元素。

在这里插入图片描述

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值