您的位置:首页 > Web前端

《剑指Offer》读书笔记05:用两个栈实现队列

2015-11-27 15:04 330 查看

题目描述

用两个栈来实现一个队列,完成队列的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;
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: