一、题目描述
给出一棵二叉树,求它的高度。二叉树的创建采用前面实验的方法。
注意,二叉树的层数是从1开始
二、输入与输出
1.输入
第一行输入一个整数t,表示有t个二叉树
第二行起输入每个二叉树的先序遍历结果,空树用字符‘0’表示,连续输入t行
1
AB0C00D00
2.输出
每行输出一个二叉树的高度
3
三、参考代码
#include <iostream>
#include <string>
using namespace std;
class TreeNode
{
public:
char ch;
TreeNode* lchild;
TreeNode* rchild;
TreeNode* parent;
TreeNode() { lchild = NULL, rchild = NULL; parent = NULL; }
};
class Tree
{
public:
TreeNode* root;
string strtree;
int point;
Tree() { };
~Tree() {};
void Create(string s)
{
point = 0;
strtree.assign(s);
root = CreateTree();
}
TreeNode* CreateTree()//递归建树
{
TreeNode* n;
char ch;
ch = strtree[point];
point++;
if (ch == '0')
{
n = NULL;
}
else
{
n = new TreeNode();
n->ch = ch;
n->lchild = CreateTree();
if (n->lchild != NULL)
{
n->lchild->parent = n;
}
n->rchild = CreateTree();
if (n->rchild != NULL)
{
n->rchild->parent = n;
}
}
return n;
}
int h(TreeNode* t)
{
if (t == NULL)
{
return 0;
}
else if (t->lchild == NULL && t->rchild == NULL)
{
return 1;
}
int lefth = h(t->lchild);
int righth = h(t->rchild);
return lefth > righth ? (lefth + 1) : righth + 1;
}
int hh()
{
return h(root);
}
};
int main()
{
int sum;
cin >> sum;
string str;
while (sum--)
{
cin >> str;
Tree t;
t.Create(str);
cout << t.hh() << endl;
}
return 0;
}
/**********************************************************************
Problem: 1053
Language: C++
Result: AC
Time:7 ms
Memory:2236 kb
**********************************************************************/