题目:用两个栈实现一个队列。队列的声明如下,请实现它的两个函数appendTail和deleteHead,分别完成在队列尾部插入节点和在队列头部删除节点的功能。
template<typename T>
class CQueue
{
public:
CQueue()
{
}
~CQueue()
{
}
void appendTail(const T& node);
T deleteHead();
private:
stack<T> stack1;
stack<T> stack2;
};
分析:
从上面可以看出,一个队列包含了两个栈stack1和stack2,因此,这道题要求我们操作这两个“先进后出”的栈实现一个“先进先出”的队列CQueue。
我们通过具体的例子来分析往该队列插入和删除元素的过程。
首先我们插入一个元素a
。因为两个栈stack1和stack2都为空,不妨先把他插入stack1,此时stack1中有元素{a},stack2为空。再加入两个元素b和c,还是插入stack1,此时stack1中有{a,b,c},c位于栈顶,stack2仍为空,如图(a)所示。
我们从队列中删除一个元素
。按照队列先入先出的原则,由于a最先进入,最先被删除的元素应该是a。元素a存储在stack1中,但并不在栈顶上,因此不能直接进行删除。如果我们把stack1中的元素逐个弹出并压入stack2,则元素在stack2中的顺序正好和原来在stack1中的顺序相反。这个时候就可以弹出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)所示。
总结
删除一个元素的步骤:
当stack2不为空时,在stack2中的栈顶元素是队列的头部,直接弹出。当stack2为空时,我们把stack1中元素逐个弹出并压入stack2,然后弹出stack2的栈顶元素。
代码
Queue.h
#pragma once
#include<stack>
#include<exception>
using namespace std;
template<typename T>
class CQueue
{
public:
CQueue()
{
}
~CQueue()
{
}
void appendTail(const T& node);
T deleteHead();
private:
stack<T> stack1;
stack<T> stack2;
};
template<typename T>
void CQueue<T>::appendTail(const T& node)
{
stack1.push(node);
}
template<typename T>
T CQueue<T>::deleteHead()
{
if (stack2.size() <= 0)
{
while (stack1.size() > 0)
{
T data = stack1.top();
stack1.pop();
stack2.push(data);
}
}
if (stack2.size() == 0)
throw new exception("queue is empty");
T head = stack2.top();
stack2.pop();
return head;
}
测试代码:
#include"Queue.h"
// ====================测试代码====================
void Test(char actual, char expected)
{
if (actual == expected)
printf("Test passed.\n");
else
printf("Test failed.\n");
}
int main()
{
CQueue<char> queue;
queue.appendTail('a');
queue.appendTail('b');
queue.appendTail('c');
char head=queue.deleteHead();
Test(head,'a');
head = queue.deleteHead();
Test(head, 'b');
queue.appendTail('d');
head = queue.deleteHead();
Test(head, 'c');
head = queue.deleteHead();
Test(head, 'd');
system("pause");
return 0;
}