面试题7:用两个栈实现队列
1.题目描述
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead,分别完成在队列尾部插入结点和在队列头部删除结点的功能。
template<typename T>
class CQueue
{
public:
CQueue(void);
~CQueue(void);
void appendTail(const T& node);
T deleteHead();
private:
stack<T> stack1;
stack<T> stack2;
};
2.题目分析
我们通过一个具体的例子来分析往该队列插入和删除元素的过程。首先插入一个元素a,不妨先把它插入到 stack1,此时 stack1中的元素有{a},stack2为空。再压入两个元素b和c,还是插入到 stack1中,此时 stack1中的元素有{a,b,c},其中c位于栈顶,而 stack2仍然是空的(如下图(a)所示)。
这个时候我们试着从队列中删除一个元素。按照队列先入先出的规则,由于a比b、c先插入到队列中,最先被删除的元素应该是a。元素a存储在 stack1中,但并不在栈顶上,因此不能直接进行删除。如果我们把stack1中的元素逐个弹出并压入 stack2,元素在 stack2中的顺序正好和原来在 stack1中的顺序相反。因此经过3次弹出 stack1和压入 stack2操作之后,stack1为空,而 stack2中的元素是{c,b,a},这个时候就可以弹出 stack2的栈顶a了。此时的 stack1为空,而 stack2的元素为{c,b},其中b在栈顶(如
下图(b)所示)。
如果我们还想继续删除队列的头部应该怎么办呢?剩下的两个元素是b和c,b比c早进入队列,因此b应该先删除。而此时b恰好又在栈顶上,因此直接弹出 stack2的栈顶即可。这次弹出操作之后, stack1中仍然为空,而 stack2为{c}(如下图(c)所示)。
接下来再插入一个元素d我们还是把它压入 stack1(如下图(d)所示),这样会不会有问题呢?我们考虑下一次删除队列的头部 stack2不为空,直接弹出它的栈顶元素c(如下图(e)所示)。而c的确是比d先进入队列,应该在d之前从队列中删除,因此不会出现任何矛盾。
从上面的分析中我们可以总结出删除一个元素的步骤:当 stack2中不为空时,在 stack2中的栈顶元素是最先进入队列的元素,可以弹出。如果stack2为空时,我们把 stack1中的元素逐个弹出并压入 stack2。由于先进入队列的元素被压到 stack1的底端,经过弹出和压入之后就处于 stack2的顶端了,又可以直接弹出。
3.code
template<typename T>
void CQueue<T>::appendTail(const T& element)/*在stack1中插入元素*/
{
stack1.push(element);
}
template<typename T>
T CQueue<T>::deleteHead()/*从stack2中删除栈顶元素*/
{
if(stack2.empty())
{
while(!stack1().empty)
{
T& data = stack1.top();
stack1.pop();
stack2.push(data);
}
}
if(stack2.empty())
{
throw new exception("queue is empty");
}
T head = stack2.top();
stack2.pop();
return head;
}