题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路:
首先,队列的特点是先进先出,栈的特点是先进后出;
两个栈,可以在队列push的时候,将数据先存放到stack1中,如果队列需要pop,则先判断stack2是否是空,如果是空,则将stack1弹出,存放到stack2中,并弹出tsack2的top,实现队列的弹出。
如果stack2不是空,则直接弹出stack2的top,实现队列弹出。
如果在stack2不为空的前提下,给队列push,直接将数据push到stack1即可,待到stack2为空,并且队列需要pop时,再将stack1放到stack2中,这样,能够保证队列既能push,又能pop。
代码:
class Solution
{
public:
void push(int node) {
stack1.push(node);
}
int pop() {
if(stack2.empty())
{
while(!stack1.empty())
{
stack2.push(stack1.top());
stack1.pop();
}
}
int a=stack2.top();
stack2.pop();
return a;
}
private:
stack<int> stack1;
stack<int> stack2;
};