终面问了一下这个很简单的题,踩坑了。原因就是之前刷题刷的飞快,也没去评论区看大家的思路,甚至当时的力扣的官方解答也不是最优解(现在改过来了)
坑:很容易想到要用两个栈模拟——不用思考,很容易想到一种:
放元素的时候直接放主栈,删除元素时都移到辅助栈,删除辅助栈顶部元素,再把剩下的搬到主栈。
求轻喷,当时我想都没想就写了这个思路,力扣过了,好了下一道。。然后面试时。。。就没有然后了。。。
刚才的想法中,一次操作里,一个元素会被移动两次(你可以选择是插入还是删除,无所谓)
优化的想法应该是一个元素在一次操作里只被移动一次
核心就是,主栈只负责接收新来的,删除元素只能从辅助栈出去。
删除时,查看辅助栈有无元素,没有的话从主栈搬过来(对应唯一一次移动操作),然后删除顶部元素。
插入时,不管三七二十一,直接往主栈里放就好了。此时辅助栈的顶部,存放的还是最早被插入的元素。
class CQueue {
stack<int> stack1,stack2;
public:
CQueue() {
while (!stack1.empty()) {
stack1.pop();
}
while (!stack2.empty()) {
stack2.pop();
}
}
void appendTail(int value) {
stack1.push(value);
}
int deleteHead() {
// 如果第二个栈为空
if (stack2.empty()) {
while (!stack1.empty()) {
stack2.push(stack1.top());
stack1.pop();
}
}
if (stack2.empty()) {
return -1;
} else {
int deleteItem = stack2.top();
stack2.pop();
return deleteItem;
}
}
};