剑指 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]
用两个栈去实现队列已经是老生常谈了,可以形象的看作是 “负负得正” 的应用。
这个题目主要需要注意一下几个细节:
- 在创建队列的时候,使用推荐的Deque而不是Stack。
- 插入“队列”的操作只需要向栈1中入栈元素即可。
- 删除元素时,如果栈2为空,则检查栈1中是否有元素。如果有元素则全部转移到栈2中,然后将栈2的栈顶元素出栈;如果栈1也没有元素则返回-1;
具体代码实现如下;
class CQueue {
Deque<Integer> stack_in;
Deque<Integer> stack_out;
public CQueue() {
stack_in = new LinkedList<Integer>();
stack_out = new LinkedList<Integer>();
}
public void appendTail(int value) {
stack_in.push(value);
}
public int deleteHead() {
int element = -1;
// 如果stack_out为空
if(stack_out.isEmpty()){
//如果此时stack_in不为空,就把stack_in里面的元素转移到stack_out里面
// 然后再取栈顶元素
if(!stack_in.isEmpty()){
while(!stack_in.isEmpty()){
int temp = stack_in.pop();
stack_out.push(temp);
}
element = stack_out.pop();
}
// 如果stack_out不为空,则直接将栈顶元素出栈
}else{
element = stack_out.pop();
}
return element;
}
}
/**
* Your CQueue object will be instantiated and called as such:
* CQueue obj = new CQueue();
* obj.appendTail(value);
* int param_2 = obj.deleteHead();
*/