用两个栈实现队列
描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路
一个栈接收push数据,另一个栈在每次push数据时都要更新保证与第一个栈的数据正好颠倒。pop时,pop第二个栈的数据,同时更新第一个栈,同样要确保与第二个栈的数据正好颠倒
代码
class Solution
{
public:
void push(int node) {
stack1.push(node);
while (stack2.size()) stack2.pop();//清空
stack<int> temp;
temp = stack1;
for (int i = 0; i < stack1.size(); i++) {//更新,颠倒
stack2.push(temp.top());
temp.pop();
}
return;
}
int pop() {
int result = stack2.top();
stack2.pop();
while (stack1.size()) stack1.pop();//清空
for (int i = 0; i < stack2.size(); i++)//更新,颠倒
stack1.push(stack2.top());
return result;
}
private:
stack<int> stack1;
stack<int> stack2;
};