DS树--二叉树高度

DS树–二叉树高度

---

题目描述

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

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

---

输入

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

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

---

输出

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

---

输入样例

1
AB0C00D00

---

输出样例

3

---

代码

#include<iostream>
#include<string>
using namespace std;

class BiTreeNode {
public:
    char data;
    BiTreeNode *LeftChild;
    BiTreeNode *RightChild;
    BiTreeNode():LeftChild(NULL),RightChild(NULL){}
};

class BiTree {
public:
    void CreateTree(string s);
    int FindDeep();
private:
    BiTreeNode *CreateTree();
    int FindDeep(BiTreeNode *t);
    int pos;
    string strTree;
    BiTreeNode *Root;
};

int BiTree::FindDeep()
{
    return FindDeep(Root);
}

int BiTree::FindDeep(BiTreeNode *t)
{
    int deep = 0;
    if(t)
    {
        int LeftChildDeep = FindDeep(t->LeftChild);
        int RightChildDeep = FindDeep(t->RightChild);
        deep = LeftChildDeep >= RightChildDeep ? LeftChildDeep+1 : RightChildDeep+1;
    }
    return deep;
}



void BiTree::CreateTree(string s)
{
    pos = 0;
    strTree.assign(s);
    Root = CreateTree();
}

BiTreeNode *BiTree::CreateTree()
{
    char ch;
    ch = strTree[pos++];
    BiTreeNode *T;
    if(ch == '0')
        T = NULL;
    else
    {
        T = new BiTreeNode();
        T->data = ch;
        T->LeftChild = CreateTree();
        T->RightChild = CreateTree();
    }
    return T;
}

int main()
{
    int n;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        string s;
        cin>>s;
        BiTree biTree;
        biTree.CreateTree(s);
        cout<<biTree.FindDeep()<<endl;
    }
    return 0;
}