DS树--二叉树高度

题目描述

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

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

输入

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

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

输出

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

解题代码

#include <iostream>
using namespace std;
class Tnode
{
    char name;
    Tnode *lChild, *rChild;

public:
    Tnode() : lChild(NULL), rChild(NULL) {}
    Tnode(char a) : name(a), lChild(NULL), rChild(NULL) {}
    friend class Tree;
};

class Tree
{
    Tnode *root;
    int height;
    void preOrder(Tnode *a, int i)
    {
        if (a)
        {
            i++;
            if (!a->lChild && !a->rChild)
            {
                if (height < i)
                    height = i;
            }
            preOrder(a->lChild, i);
            preOrder(a->rChild, i);
        }
    }
    void midOrder(Tnode *a)
    {
        if (a)
        {
            midOrder(a->lChild);
            cout << a->name;
            midOrder(a->rChild);
        }
    }
    void finOrder(Tnode *a)
    {
        if (a)
        {
            finOrder(a->lChild);
            finOrder(a->rChild);
            cout << a->name;
        }
    }
    void CreatTree(Tnode *&a); //这里*&应该理解为引用指针类型

public:
    Tree() : root(NULL), height(1) {}
    void CreatTree()
    {
        CreatTree(root);
    }
    void preOrder()
    {
        int i = 0;
        preOrder(root, i);
        cout << height << endl;
    }
    void midOrder()
    {
        midOrder(root);
        cout << endl;
    }
    void finOrder()
    {
        finOrder(root);
        cout << endl;
    }
    ~Tree()
    {
        delete root;
    }
};

void Tree::CreatTree(Tnode *&a)
{
    char c;
    cin >> c;
    if (c != '0')
    {
        Tnode *t = new Tnode(c); //此处可简化为a = new Tnode(c); 下面一行就不需要了
        a = t;
        CreatTree(a->lChild);
        CreatTree(a->rChild);
    }
    else
        a = NULL;
}

int main()
{
    int t;
    cin >> t;
    while (t--)
    {
        Tree mytree;
        mytree.CreatTree();
        mytree.preOrder();
    }
    return 0;
}