题目:
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路:
队列先进先出,栈先进后出。用两个栈实现队列的先进先出。
一个栈用于插入队列元素,一个栈用于删除队列元素:
插入:队列中依次插入元素a,b,c,即在栈stack1中插入a,b,c;
删除:队列中删除元素顺序为a,b,c,栈stack1中删除元素顺序为c,b,a。
因为栈为先入后出,将stack1中的元素依次弹出到栈stack2中,stack2中的元素顺序为c,b,a,这时删除stack2元素顺序为a,b,c,与队列删除顺序一致。
栈2中有元素,直接删除;栈2中没有元素则向栈2中插入栈1中的所有元素,再删除。
测试用例:
1.空队列的插入删除
2**.非空队列**的插入删除
3.连续删除元素直到队列为空
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;
};
用两个队列实现一个栈:P71
将不需要删除的元素移动到另一个队列中,删除本队列中的最后一个元素。