用两个栈实现队列

用两个栈实现队列

描述

用两个栈来实现一个队列，完成队列的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;
};