剑指 Offer 09. 用两个栈实现队列
一、题目:
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
示例 1:
输入:
[“CQueue”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[3],[],[]]
输出:[null,null,3,-1]
二、思路、代码和结果:
思路:
首先要了解一下栈和队列的特性:
栈是先进后出(先进栈的元素后出栈),队列是先进先出(先进队的元素先出队);
实现的方法是:
1、元素进队:直接进栈1即可,栈1元素和队列元素一致;
2、元素出队:按照队列的特性,应该出栈1中最早进入的元素,这个时候,我们使用栈2对出栈1的元素进行入栈操作,也就是将栈1的元素翻转,此时栈2中的元素出栈就和栈1作为队列出队一样了。
class CQueue {
stack<int>st1,st2;
public:
CQueue() {
}
void appendTail(int value) {//队尾插入
st1.push(value);
}
int deleteHead() {//队头删除
//栈2为空
if(st2.empty()){
//将栈1的元素出栈后再进入栈2
while(!st1.empty()){
st2.push(st1.top());
st1.pop();
}
}
//如果栈2中有元素要先把栈2中的元素出栈完了,再把1中的倒腾过来
if(!st2.empty()){
int a = st2.top();
st2.pop();
return a;
}
else
return -1;
}
};
/**
* Your CQueue object will be instantiated and called as such:
* CQueue* obj = new CQueue();
* obj->appendTail(value);
* int param_2 = obj->deleteHead();
*/
结果:
怕什么真理无穷,进一步有进一步的欢喜!