题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
给定函数头
class Solution
{
public:
void push(int node) {
}
int pop() {
}
private:
stack<int> stack1;
stack<int> stack2;
};
思路:
在执行push时,令stack2为空,将待存元素压入stack1的栈顶;
在执行pop时,令stack1为空,stack2的栈顶元素即为待pop元素;
用容器实现起来特别简单。
代码
class Solution
{
public:
//stack1用于存储,stack2用于临时存储stack1的元素,以完成出队
void push(int node) {
while (!stack2.empty()) { //执行push时,令stack2为空,将待存元素压入stack1的栈顶;
stack1.push(stack2.top());
stack2.pop();
}
stack1.push(node);
}
int pop() {
while(!stack1.empty()) { //执行pop时,令stack1为空,stack2的栈顶元素即为待pop元素;
stack2.push(stack1.top());
stack1.pop();
}
int temp = stack2.top();//取出stack2栈顶元素,存入临时变量temp,待返回。
stack2.pop(); //别忘了销毁哦
return temp;
}
private:
stack<int> stack1;
stack<int> stack2;
};