剑指 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class CQueue {
LinkedList<Integer> stack1; //先声明出两个对象
LinkedList<Integer> stack2;
public CQueue() {
// 为两个对象分配空间
stack1 = new LinkedList<>();
stack2 = new LinkedList<>();
}
public void appendTail(int value) {
stack1.push(value); //append 推进栈1 一个元素
}
public int deleteHead() {
if(stack2.isEmpty()){ //栈2 要是为空的话 紧接着判断栈1
if(stack1.isEmpty()) return -1;
while(!stack1.isEmpty()){ //要是栈1不为空的话 那就将栈1的元素全都移入到栈2
stack2.push(stack1.pop());
}
return (int)stack2.pop();
}else{ //要是栈2不为空的话 直接弹出栈2的元素
return (int)stack2.pop();
}
}
}