剑指 Offer 09. 用两个栈实现队列
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
class CQueue {
public:
stack<int> s1;
stack<int> s2;
CQueue() {
}
void appendTail(int value) {
s1.push(value);
}
int deleteHead() {
if(s1.empty()){
return -1;
}
while(!s1.empty()){
s2.push(s1.top());
s1.pop();
}
int q = s2.top();
s2.pop();
while(!s2.empty()){
s1.push(s2.top());
s2.pop();
}
return q;
}
};
虽然很慢但是比较好想,去年考研数据结构大题就是这个:
s1栈用于实现add的功能,s2栈复杂很多,用于实现delete的功能。
为了用两个栈来模拟队列,即先将数据先进后出顺序压入一个栈,再依次弹出压入第二个栈,此时第二个栈再弹出就符合队列先进先出的逻辑了。
为了维护真正意义的队列,在每次s1栈清空并完全转入s2栈后,记录s2的top,然后再将top弹出,将剩下的元素依次弹出并压回s1,此时s2变为空栈,s1保存删去队列头之后的剩余元素。又恢复到了可以接受命令的初始状态。
特殊情况处理:在s1为空的情况下还要delete,此时s1无元素弹出并压入s2,所以s2直接返回-1。
官方题解的思想是不用每次s2得到top后再将元素压回s1,而是利用if判断,在s2尚未空时,若delete仍从s2中简单top+pop即可(因为s2中元素仍然符合队列结构),此时即使add也只是压入s1,不影响s2。直到s2为空,此时又该将s1中元素调入s2了。
if (outStack.empty()) {
if (inStack.empty()) {
return -1;
}
in2out();//用于将s1调入s2
}