题链:https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof/
思路1
本人模拟法,时间最坏
O
(
N
2
)
O(N^2)
O(N2),不建议使用。
每次删除时,先把当前栈所有元素放入新的栈,把新栈栈顶元素弹出。
class CQueue {
public:
stack<int> s1,s2;
CQueue() {
}
void appendTail(int value) {
s1.push(value);
}
int deleteHead() {
if(s1.empty()) return -1;
while(!s2.empty()) s2.pop();
while(!s1.empty()){
s2.push(s1.top());
s1.pop();
}
int x = s2.top();
s2.pop();
while(!s2.empty()){
s1.push(s2.top());
s2.pop();
}
return x;
}
};
/**
* Your CQueue object will be instantiated and called as such:
* CQueue* obj = new CQueue();
* obj->appendTail(value);
* int param_2 = obj->deleteHead();
*/
思路2
官方题解:维护两个栈,第一个栈s1只放放入新元素,第二个栈s2只取出栈顶元素。实现方式为:
(1)放入元素时,放入栈s1。
(2)取出元素时,首先查看栈s2是否为空。若为空,就将栈s1全部元素放入栈s2。如果栈s2仍为空,返回-1;否则就把栈顶元素弹出。
这样最坏复杂度仅为
O
(
2
N
)
O(2N)
O(2N)。
class CQueue {
public:
stack<int> s1,s2;
CQueue() {
}
void appendTail(int value) {
s1.push(value);
}
int deleteHead() {
if(s2.empty()){
while(!s1.empty()){
s2.push(s1.top());
s1.pop();
}
}
if(s2.empty()) return -1;
int x=s2.top();
s2.pop();
return x;
}
};
/**
* Your CQueue object will be instantiated and called as such:
* CQueue* obj = new CQueue();
* obj->appendTail(value);
* int param_2 = obj->deleteHead();
*/