用两个栈实现队列
题目描述:
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
代码如下
此代码为优化过后的代码,如果按平常的思维来说,一个用来输入,一个用来删除,这样的话,两个栈会倒来倒去,但其实我们可以发现,例如第一次我们导入过后,发现s2里面就是正常顺序底部元素,如果要删除的话,其实可以直接在这里开始删除,不必再把s2导入到s1中去了,这个时候如果还有要加入的直接放进s1中,暂存。这个过程直到s2中没有元素了,就在进行一次导入s2,如此反复,效率相较于一般更高一些。
class CQueue {
Stack<Integer> s1;
Stack<Integer> s2;
public CQueue() {
s1 = new Stack<>();
s2 = new Stack<>();
}
public void appendTail(int value) {
s1.push(value);
}
public int deleteHead() {
//s2是否为空
if(s2.isEmpty()) {
//s1为空的话则为空,返回-1
if(s1.isEmpty()) {
return -1;
}
//然后将s1全部转移到s2上
while(!s1.isEmpty()) {
s2.push(s1.pop());
}
}
return s2.pop();
}
}