1021: 二叉链表(2)
1021: 二叉链表(2)
Description
递归算法计算二叉树的叶子个数和高度。请完成如三个算法:
int CountLeaf(BiNode<T> *); // 递归算法计算叶子数
int Depth(BiNode<T> *); // 递归算法计算高度
BiNode<DataType> *GetRoot(); // 获取私有量root的值
部分已知代码如下,请在此基础上补充完善代码:
#include <iostream>
#include <string>
using namespace std;
template <typename DataType>
struct BiNode // 二叉树的结点结构
{
DataType data;
BiNode<DataType> *lchild, *rchild;
};
// 二叉树类,实现计算高度、叶子数
template <typename DataType>
class BiTree
{
public:
BiTree(); // 构造函数,初始化一棵二叉树,其前序序列由键盘输入
~BiTree(); // 析构函数,释放二叉链表中各结点的存储空间
int CountLeaf(BiNode<DataType> *); // 递归方法计算叶子数
int Depth(BiNode<DataType> *); // 递归方法计算高度
BiNode<DataType> *GetRoot(); // 获取私有量root的值
int CountLeaf(); // 调用上面的递归算法计算叶子
int Depth(); // 调用上面的递归算法计算高度
private:
BiNode<DataType> *root; // 指向根结点的头指针
void Creat(BiNode<DataType> *&bt); // 被构造函数调用,递归方式生成二叉树
void Release(BiNode<DataType> *&bt); // 被析构函数调用
};
// 定义类中的成员函数
// 构造函数:Creat利用创建二叉树
template <typename DataType>
BiTree<DataType>::BiTree()
{
Creat(root);
}
// 功 能:递归方法创建一棵二叉树,由构造函数调用
template <typename DataType>
void BiTree<DataType>::Creat(BiNode<DataType> *&bt)
{
DataType ch;
cin >> ch;
if (ch == "#")
bt = nullptr; // 创建结点值为字符串的二叉树
else
{
bt = new BiNode<DataType>; // 生成一个结点
bt->data = ch;
Creat(bt->lchild); // 递归建立左子树
Creat(bt->rchild); // 递归建立右子树
}
}
// 功 能:析构函数,释放二叉链表中各结点的存储空间
template <typename DataType>
BiTree<DataType>::~BiTree() // 析构函数不能带参数
{
Release(root);
}
// 功 能:释放二叉树的存储空间,析构函数调用
template <typename DataType>
void BiTree<DataType>::Release(BiNode<DataType> *&bt)
{
if (bt != nullptr)
{
Release(bt->lchild); // 释放左子树
Release(bt->rchild); // 释放右子树
delete bt;
}
}
template <typename DataType>
int BiTree<DataType>::Depth()
{
return Depth(root);
}
template <typename DataType>
int BiTree<DataType>::CountLeaf()
{
return CountLeaf(root);
}
// 请在下面补充相关算法的实现
int main()
{
BiTree<string> mybit;
cout << mybit.CountLeaf() << " " << mybit.Depth() << endl;
BiNode<string> *broot;
broot = mybit.GetRoot();
cout << mybit.CountLeaf(broot) << " " << mybit.Depth(broot) << endl;
return 0;
}
Input
Output
Sample Input
Li Sun # Zhao Zhou # # Wang # # Qian # #
Sample Output
3 4
3 4