最初重温了下数据结构的栈和队列的基本使用,想要在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)。需要使用两个栈存储已有的元素。