题目描述
给出一棵二叉树,求它的高度。
注意,二叉树的层数是从1开始
输入
第一行输入一个整数t,表示有t个二叉树
第二行起输入每个二叉树的先序遍历结果,空树用字符‘0’表示,连续输入t行
输入样例:
1
AB0C00D00
输出
每行输出一个二叉树的高度
输出样例:
3
代码
#include <iostream>
using namespace std;
struct Node{
char data;
Node *left;
Node *right;
};
class Tree{
public:
Node *root;
void create(Node *&t){
char c;
cin >> c;
if(c == '0'){
t = NULL;
}else{
t = new Node;
t->data = c;
create(t->left);
create(t->right);
}
}
int height(Node *t){ //递归求二叉树高度
if(t == NULL){
return 0;
}else if(t->left == NULL && t->right == NULL) { //叶子结点 return 1
return 1;
}
int lheight = height(t->left);
int rheight = height(t->right);
if(lheight > rheight){
return lheight+1;
}else{
return rheight+1;
}
}
};
int main()
{
int t;
cin >> t;
while(t--){
Tree tree;
tree.create(tree.root);
int height = tree.height(tree.root);
cout << height << endl;
}
return 0;
}