题目:
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
题目解读:
要求:使用俩个栈模拟队列
栈是先进后出的,我们使用俩个栈,其中栈A入栈,先进后出地将元素装入栈B,那么此时栈B中的元素就是逆序的,然后栈B中的元素出栈元素,先进后出,相当于将A中的元素实现了先进先出。
例如元素进入顺序 :123456
A栈中元素顺序:123456;先进后出:654321
B栈中元素顺序:654321;先进后出:123456
题解:
class CQueue {
public:
stack<int> a,b; //定义俩个栈 a入栈 b逆序装a,然后出栈
CQueue() {
}
void appendTail(int value) {
a.push(value);//将元素入栈
}
int deleteHead() {
if(!b.empty())
{
int tmp=b.top();
b.pop();//b的栈顶元素出栈
return tmp;
}
else
{
if(a.empty())
{
return -1;
}
else{
while(!a.empty()) //这里要将A中所有元素都加入到栈B中,才能进最开始的元素出栈
{
int tmp = a.top();
a.pop();
b.push(tmp);
}
int tmp = b.top();
b.pop();
return tmp;
}
}
}
};
/**
* Your CQueue object will be instantiated and called as such:
* CQueue* obj = new CQueue();
* obj->appendTail(value);
* int param_2 = obj->deleteHead();
*/
```cpp
在这里插入代码片
## 复杂度分析:
以下分析仅满足添加 N 个元素并删除 N 个元素,即栈初始和结束状态下都为空的情况。
**时间复杂度**: appendTail()函数为 O(1)O(1) ;deleteHead() 函数在 NN 次队首元素删除操作中总共需完成 NN 个元素的倒序。
**空间复杂度** O(N) : 最差情况下,栈 A 和 B 共保存 NN 个元素。
**题目地址:**
https://leetcode-cn.com/leetbook/read/illustration-of-algorithm/5d3i87/
来源:力扣(LeetCode)