Leetcode 剑指 Offer 09. 用两个栈实现队列
题目
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 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]
思路
- 双栈实现队列的思想就是先定义2个栈, 一个栈stack1专门插入元素, 一个栈stack2专门弹出元素
- 插入就是直接插入到stack1即可
- 删除首先要判断是否没有元素, 没有元素则返回-1, stack2空的时候则把stack1的元素全部导入stack2, stack2非空的时候则直接从stack2栈顶返回元素即可
代码 —— c++
class CQueue {
private:
stack<int> stack1, stack2;
public:
CQueue() {
}
void appendTail(int value) {
stack1.push(value);
}
int deleteHead() {
if (stack1.empty() && stack2.empty()) return -1;
if (stack2.empty()) {
while (!stack1.empty()) {
stack2.push(stack1.top());
stack1.pop();
}
}
auto res = stack2.top();
stack2.pop();
return res;
}
};
/**
* Your CQueue object will be instantiated and called as such:
* CQueue* obj = new CQueue();
* obj->appendTail(value);
* int param_2 = obj->deleteHead();
*/