题目描述
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 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 次调用
思路:两个栈实现队列,一个栈用来完成队列的在队尾添加功能,另一个栈完成队列的队首删除功能
这个地方有点绕,需要注意的是:当stack2不为空时,队列中的元素都在stack2中(stack2中都是待删除的元素),无论stack1中是否有元素都从2中弹出元素
频繁的插入和删除首选LinkedList,通常情况下不推荐使用Vector和及其子类Stack
解决方法
class CQueue {
LinkedList<Integer> stack1, stack2;
public CQueue() {
stack1 = new LinkedList<>();
stack2 = new LinkedList<>();
}
public void appendTail(int value) {
stack1.addLast(value);
}
public int deleteHead() {
//分三步进行判断:1.stack2不为空,直接返回最后一个元素
//2.stack1为空,直接返回-1;3.stack1不为空,stack2存放stack1的倒序,并输出stack2的最后一个元素
if(!stack2.isEmpty()){
return stack2.removeLast();
}
if(stack1.isEmpty()){
return -1;
}
while(!stack1.isEmpty()){
stack2.addLast(stack1.removeLast());
}
return stack2.removeLast();
//这个的理解:判断2是否为空,1是否为空;
//if(stack2.isEmpty()){
// if(stack1.isEmpty()){
// return -1;
// }else{
// while(!stack1.isEmpty()){
// stack2.addLast(stack1.removeLast());
// }
// }
//}
return stack2.removeLast();
}
}
/**
* Your CQueue object will be instantiated and called as such:
* CQueue obj = new CQueue();
* obj.appendTail(value);
* int param_2 = obj.deleteHead();
*/