链接:剑指 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]
·
提示: 1 <= values <= 10000 最多会对 appendTail、deleteHead 进行 10000 次调用
·
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题目里面关于CQuene
并没有说明具体含义。
从官方解答来看,CQuene
函数代表清空队列操作。因此这里采用一个while循环进行操作。
建立两个栈s1、s2,输入到s1。
删除头结点时:
- s1、s2均为空:空队列,返回-1。
- s1有元素、s2为空:s1元素依次出栈压入s2,完成操作后,s2推出栈顶元素。
- s2有元素:s2推出栈顶元素。
class CQueue {
public:
stack<int> s1,s2;
CQueue() {
while(!s1.empty()) s1.pop();
while(!s2.empty()) s2.pop();
}
void appendTail(int value) {
s1.push(value);
}
int deleteHead() {
if(s2.empty() && s1.empty())
return -1;
else if(s2.empty() && !s1.empty()){
while(!s1.empty()){
s2.push(s1.top());
s1.pop();
}
}
int tmp=s2.top();
s2.pop();
return tmp;
}
};
运行结果: