IMWeb提升营Day1 | 训练题5： 用两个栈实现队列

题目描述

用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型。

思路

/*
* 思路：两个栈实现一个队列
* --入队列：直接进栈1
* --出队列：从栈2出元素，判断队列2是否为空，如果已经弹空了点话就从栈1把元素都弹到栈2里来，然后再从栈2弹出元素
*/
class Solution
{
public:
void push(int node) {
stack1.push(node);
}

int pop() {
int a;
if(stack2.empty()){
while(!stack1.empty()){
a = stack1.top();
stack2.push(a);
stack1.pop();
}
}

a = stack2.top();
stack2.pop();
return a;
}

private:
stack<int> stack1;
stack<int> stack2;
};

举例一返二：

两个队列实现一个栈的思路：

–入栈：直接进队列1

–出栈：判断队列1里是不是只有一个元素，是的话直接从队列1里面出，如果是多于一个元素的话，就把之前先进来的放到队列2里面去，把最后一个进队列1的元素弹出，实现出栈的效果，然后再把队列2的元素都放回队列1.