题目:用两个栈实现一个队列。队列的声明如下,请实现它的两个函数appendTail和deleteHead,分别完成在队列尾部插入结点和在队列头部删除结点的功能。
类的定义如下:
template <typename T> class CQueue
{
public:
CQueue(void);
~CQueue(void);
// 在队列末尾添加一个结点
void appendTail(const T& node);
// 删除队列的头结点
T deleteHead();
public:
stack<T> stack1;
stack<T> stack2;
};
解题思路
看到这道题,首先应该思考这两种数据结构有什么特点以及它们之间的联系。栈的特点是“先进后出”,队列的特点是“先进先出”,让我们用两个“先进后出”的栈来实现一个“先进先出”的队列,假如那么我们先把数据压入第一个栈,此时它的出栈顺序与想要实现的队列的出队顺序是相反的,如果再把这个栈里面的元素依次压入第二个栈,相当于对一组数据进行了两次倒序,此时第二个栈的出栈顺序就与数据出队的顺序一致。
删除节点时,首先判断stack2是否为空,如果为空,则将stack1栈中的数据出栈并压入栈stack2中,操作完成后如果stack2依旧为空,则结束程序。
上述解题步骤实现主要代码如下:
template<typename T> T CQueue<T>::deleteHead()
{
if(stack2.empty())
{
while(!stack1.empty())
{
//T& data = stack1.top();
stack2.push(stack1.top());
stack1.pop();
}
}
if(stack2.empty())
throw new exception("queue is empty");
T head = stack2.top();
stack2.pop();
return head;
}
由于为模板类,所以声明和定义要放在同一个文件中。类的具体声明定义如下:
#pragma once
#include <stack>
#include <exception>
using namespace std;
template <typename T> class CQueue
{
public:
CQueue(void);
~CQueue(void);
// 在队列末尾添加一个结点
void appendTail(const T& node);
// 删除队列的头结点
T deleteHead();
public:
stack<T> stack1;
stack<T> stack2;
};
template <typename T> CQueue<T>::CQueue(void)
{
}
template <typename T> CQueue<T>::~CQueue(void)
{
}
template<typename T> void CQueue<T>::appendTail(const T& element)
{
stack1.push(element);
}
template<typename T> T CQueue<T>::deleteHead()
{
if(stack2.empty())
{
while(!stack1.empty())
{
//T& data = stack1.top();
stack2.push(stack1.top());
stack1.pop();
}
}
if(stack2.empty())
throw new exception("queue is empty");
T head = stack2.top();
stack2.pop();
return head;
}
测试用例1
// ====================测试代码====================
void Test(char actual, char expected)
{
if (actual == expected)
printf("Test passed.\n");
else
printf("Test failed.\n");
}
int main(int argc, char* argv[])
{
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');
queue.appendTail('e');
head = queue.deleteHead();
Test(head, 'd');
head = queue.deleteHead();
Test(head, 'e');
system("pause");
return 0;
}