题干
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 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]
想法
两个栈,stack1用来直接push进去,一旦需要删除了,如果另一个栈res为空,就讲stack1里的所有元素push进去。然后删除栈顶即可。
这都很简单
有一种需要注意的情况:
如果两个都不为空,需要删除呢?
那就直接删除res的栈顶,相当于把队列存了两节,当res空了再触发下一次更新。
不然时间复杂度感人。
Java代码
package daily;
import java.util.Stack;
public class CQueue {
Stack<Integer> stack1;
Stack<Integer>res;
public CQueue() {
stack1=new Stack();
res=new Stack();
}
//加尾巴直接加
public void appendTail(int value) {
stack1.push(value);
}
//res空,那就stack1直接倒过去,然后返回栈顶。否则直接返回栈顶
public int deleteHead() {
if(res.isEmpty()){
while(!stack1.isEmpty()){
res.push(stack1.pop());
}
}
if(res.isEmpty()){
return -1;
}
return res.pop();
}
}