题目描述
用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 appendTail 和 deleteHead ，分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素，deleteHead 操作返回 -1 )
示例：
输入：
["CQueue","appendTail","deleteHead","deleteHead"]
[[],[3],[],[]]
输出：[null,null,3,-1]
输入：
["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]
[[],[],[5],[2],[],[]]
输出：[null,-1,null,null,5,2]
说明：
- 1 <= values <= 10000
- 最多会对 appendTail、deleteHead 进行 10000 次调用
题目链接： https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof/
思路
栈是后进先出的数据结构，例如，依次输入1,2,3，则第一个弹出的元素为 3。我们想用栈来实现队列，也就是依次输入1,2,3，第一个输出的元素为 1。我们可以借助两个栈来实现队列，设两个栈分别为 s1 和 s2。首先将元素压入 s1，当要弹出元素时，我们先将 s1 中的元素依次弹出并依次压入 s2 中，这时 s2 中存放的是 s1 中元素的逆序，则弹出 s2 的栈顶元素即可。例如，将 [1,2,3] 压入 s1 中，s1 中从栈顶到栈底的元素为 [3,2,1]，然后将 s1 中的元素弹出并压入 s2，则 s2 中从栈顶到栈底的元素为 [1,2,3]，此时弹出 s2 中的栈顶元素即可。
总结：
s1 只用来插入，s2 只用来删除。
插入：
- 将元素放进 s1 中；
删除：
- 如果 s2 不空，则将 s2 中的元素弹出并返回；
- 否则，如果 s1 不空，则将 s1 中的元素弹出并压入 s2，然后将 s2 中的元素弹出并返回；
- 否则，说明 s1、s2 均为空，返回 -1；
代码如下：
class CQueue {
private:
stack<int> s1;
stack<int> s2;
public:
CQueue() {
}
void appendTail(int value) {
s1.push(value);
}
int deleteHead() {
if(!s2.empty()){
int t = s2.top(); s2.pop();
return t;
}
else if(!s1.empty()){
while(!s1.empty()){
int t = s1.top(); s1.pop();
s2.push(t);
}
int t = s2.top(); s2.pop();
return t;
}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();
*/
- 时间复杂度：O(n)
- 空间复杂度：O(n)