剑指 Offer 09. 用两个栈实现队列
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
来源:力扣(LeetCode)
思路:
维护两个栈stack1,stack2,第一个栈支持插入操作,第二个栈支持删除操作。
第一个栈只负责插入操作,在进行删除操作时,首先判断第二个栈是否为空,如果为空,当第一个栈不为空的情况下,把第一个栈的栈顶元素加入到第二个栈中。如果此时第二个栈依旧为空,则返回-1。要执行删除操作的时候直接弹出第二个栈的元素返回即可。
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 result = stack2.top();
stack2.pop();
return result;
}
}
};
/**
* Your CQueue object will be instantiated and called as such:
* CQueue* obj = new CQueue();
* obj->appendTail(value);
* int param_2 = obj->deleteHead();
*/