6-5 E. DS树--二叉树高度

题目描述

给出一棵二叉树，求它的高度。

注意，二叉树的层数是从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;
}