剑指offer 09.用两个栈实现队列
题目描述:
用两个栈实现一个队列。
队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,
分别完成在队列尾部插入整数和在队列头部删除整数的功能。
(若队列中没有元素,deleteHead 操作返回 -1 )
示例1:
输入:
[“CQueue”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[3],[],[]]
输出:
[null,null,3,-1]
示例2:
输入:
[“CQueue”,“deleteHead”,“appendTail”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[],[5],[2],[],[]]
输出:
[null,-1,null,null,5,2]
考虑到这两个示例不太好懂,这里稍稍解释一下。
输入第一行的字符串是指调用的函数,CQueue是构造函数,deleteHead指删除头元素,appendTail指添加尾元素。
第二行指各函数对应的参数。
输出则是各函数的返回值。
思路与解法(一)——一个栈用于添加,一个栈用于删除
我们知道栈是先入后出FILO的,若仅有一个栈,想直接取出队列的头元素是不可能的,因为它在栈底。
那么两个栈怎么做呢?这里基于的原理是,将Stack1中的元素依次先pop再push进Stack2,可以实现倒序。
原本在栈底的元素变为栈顶。
这样Stack1做添加尾元素,Stack2做删除头元素即可。
Java实现如下:
class CQueue {
Stack<Integer> s1, s2;
public CQueue() {
s1 = new Stack<>();
s2 = new Stack<>();
}
public void appendTail(int value) {
s1.push(value);
}
public int deleteHead() {
if(s2.isEmpty()){
if(s1.isEmpty()){
return -1;
}
while(!s1.isEmpty()){
s2.push(s1.pop());
}
}
return s2.pop();
}
}
复杂度分析
时间复杂度:appendTail函数时间复杂度为O(1)。deleteHead删除n次头元素需要对n个元素实现倒序。
空间复杂度:O(n),最坏情况下两个栈保存n个元素。