题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。
队列中的元素为int类型。
解题思路
- 队列的特点先进先出,栈的特点后进先出。
- 假如序列a,b,c压入栈stack1,在stack1中出栈,以顺序c,b,a压入stack2,那么stack2出栈时,顺序变为a,b,c,基本满足队列要求。
- 在步骤2中,假如stack2中的a,b出栈,留下c,此时在stack1中压入d,e。此时能不能把d,e从stack1中弹出,压入stack2呢?从整个入栈顺序来看,下一个出栈的应该是c,若e,d压入stack2,下一个出栈的是d,不符合队列弹出顺序。
- 从步骤3可以得出,stack2中有元素时,先弹出。当stack2为空时,才能从stack1弹出元素压入stack2.
参考代码
class Solution
{
public:
void push(int node) {
stack1.push(node);
}
int pop() {
if ( stack2.empty() ) {//stack2为空时,才从stack1压入元素
while ( !stack1.empty() ) {
int tmp = stack1.top();//top()为取栈顶元素值
stack2.push(tmp);
stack1.pop();//pop()为弹出(删除)栈顶元素。
//top、pop容易傻傻分不清楚
}
}
if ( stack2.empty() ) {
//throw new exception("Queue is empty!");
}
int theResult = stack2.top();
stack2.pop();
return theResult;
}
private:
stack<int> stack1;
stack<int> stack2;
};