DS树--二叉树高度

目录

题目描述

思路分析

AC代码

---

题目描述

给出一棵二叉树，求它的高度。二叉树的创建采用前面实验的方法。

注意，二叉树的层数是从1开始

输入

第一行输入一个整数t，表示有t个二叉树

第二行起输入每个二叉树的先序遍历结果，空树用字符‘0’表示，连续输入t行

输出

每行输出一个二叉树的高度

输入样例1

1
AB0C00D00

输出样例1

3

思路分析

首先把树给建立起来，递归建立树的每个节点，先建立数据，再递归建立左子树，然后递归建立右子树，递归结束的条件是到了字符串末尾或者遇到字符0。

我一开始的想法是，计算出每个节点的深度，然后找出最大的深度，后来出了点问题，在我的学长的光芒下，用三行代码算出了树的高度。

递归求解树的高度：

如果节点为空，返回0，否则返回左子树和右子树的高度的最大者加一。

膜拜大佬。

AC代码

#include <iostream>
#include <string>
using namespace std;
class BiTreeNode {
public:
    char  data;					//数据域
    BiTreeNode  *leftChild,  *rightChild;	//左右子树指针
    BiTreeNode():leftChild(NULL), rightChild(NULL){}
    ~BiTreeNode() {}
};

class BiTree {
private:
    BiTreeNode *root;	//根结点指针
    string sTree;		//建树字符串
    int pos;			//标识建树字符串的当前字符位置
    BiTreeNode * CreateTree();//建树私有函数
public:
    BiTree():root(NULL) {};
    void Create(string vArray);	//建树公有接口，参数是特定的先序遍历字符串
    int CountHeight(BiTreeNode*T) {
        if(T==NULL)
            return 0;
        return max(CountHeight(T->leftChild), CountHeight(T->rightChild))+1;
    }
    void Show(){
        cout << CountHeight(root) << endl;
    }
};
void BiTree::Create(string vArray)
{	pos=0;
    sTree.assign(vArray);	//把参数保存到内部字符串
    root = CreateTree();	//建树成功后root指向根结点
}
BiTreeNode* BiTree::CreateTree() {
    if(pos==sTree.size()||sTree[pos]=='0'){
        pos++;
        return NULL;
    }
    BiTreeNode*T=new BiTreeNode();
    T->data=sTree[pos++];
    T->leftChild=CreateTree();
    T->rightChild=CreateTree();
    return T;
}
int main()
{	int t;
    string vArray;
    cin>>t;
    while(t--)
    {	cin>>vArray;
        BiTree myTree;
        myTree.Create(vArray);
        myTree.Show();
    }
    return 0;
}