5.用两个栈实现队列
题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路
创建两个栈stack1和stack2,使用两个“先进后出”的栈实现一个“先进先出”的队列。
流程示意图如下:
具体思路:当stack2中不为空时,在stack2中的栈顶元素是最先进入队列的元素,可以弹出。如果stack2为空时,我们把stack1中的元素逐个弹出并压入stack2。由于先进入队列的元素被压倒stack1的栈底,经过弹出和压入之后就处于stack2的栈顶,有可以直接弹出。如果有新元素d插入,我们直接把它压入stack1即可。
代码实现
class Solution
{
public:
void push(int node) {
stack1.push(node); //队列的push操作
}
int pop() { //队列的pop操作,按顺序pop
if ( stack2.empty())
{
while (stack1.size()>0)
{
int data = stack1.top();
stack1.pop();
stack2.push(data);
}
}
int pop_element = stack2.top();
stack2.pop();
return pop_element;
}
private:
stack<int> stack1;
stack<int> stack2;
};