题目描述
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 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]
提示: 1 <= values <= 10000 最多会对 appendTail、deleteHead 进行 10000 次调用
方法一(双栈)
1.解题思路
用一个栈(记作栈1)来存储添加到队列尾部的元素,用另一个栈(记作栈2)来存储即将弹出队头的元素,如何保证元素的先进先出呢,我们可以在栈2为空的时候,只要栈1有元素,就全部弹到栈2,由于反转了两次,那么出队的时候,就会优先弹出先进入栈1的元素。那什么时候队列为空呢,当栈1和栈2都为空的时候,队列就为空了。
2.代码实现
class CQueue {
ArrayDeque<Integer> s1;
ArrayDeque<Integer> s2;
public CQueue() {
s1=new ArrayDeque<>();
s2=new ArrayDeque<>();
}
public void appendTail(int value) {
s1.push(value);
}
public int deleteHead() {
if(s2.isEmpty()&&s1.isEmpty()){
return -1;
}
if(s2.isEmpty()){
while(!s1.isEmpty()){
s2.push(s1.pop());
}
}
return s2.pop();
}
}
3.复杂度分析
- 时间复杂度:appendTail直接入栈1,显然复杂度是O(1),deleteHead需要先入栈2,再弹出,但是每个元素只插入和弹出1次,所以均摊下来的复杂度为O(1)
- 空间复杂度:需要两个栈来存储所有出入队列的元素,所以复杂度为O(n)
剑指offer全集入口: 请戳这里