用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
提交代码:
class Solution
{
public:
//stack1 作为中间转运容器
void push(int node) {
while (!stack2.empty())
{
stack1.push(stack2.top());
stack2.pop(); //删除
}
stack2.push(node);
while (!stack1.empty())
{
stack2.push(stack1.top());
stack1.pop();
}
}
int pop() {
if (stack2.empty())
{
return 0;
}
int out = stack2.top(); //不能直接返回
stack2.pop(); //从stack中删除
return out;
}
private:
stack<int> stack1;
stack<int> stack2;
};
测试代码:
int main()
{
Solution *s = new Solution();
for (int i=0;i<10;i++)
{
s->push(i);
if (i==5)
{
s->pop();
}
}
system("pause");
return 0;
}