描述
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
示例
输入:
[“CQueue”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[3],[],[]]
输出:
[null,null,3,-1]
输入:
[“CQueue”,“deleteHead”,“appendTail”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[],[5],[2],[],[]]
输出:
[null,-1,null,null,5,2]
思路
栈是先入后出,而队列是先入先出,显然出栈顺序和出队顺序是相反的。
那么,可以用两个栈 A,B,只从 A 中放入新的元素,只从 B 中获取元素。当 A 中放入若干个元素后,需要获取一个元素时,此时 B 为空,那么就将 A 中的所有元素按出栈顺序加入 B 中,再从 B 中弹出栈顶元素,此时获取的元素就是按 FIFO 顺序的。再次获取元素时,B 不为空,则直接获取即可。
代码
class CQueue {
private Stack<Integer> stack1 = null;
private Stack<Integer> stack2 = null;
public CQueue() {
stack1 = new Stack<>();
stack2 = new Stack<>();
}
public void appendTail(int value) {
stack1.push(value);
}
public int deleteHead(){
if (stack2.isEmpty()) {
if (stack1.isEmpty())
return -1;
while (!stack1.isEmpty()) {
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
}