DS树--二叉树高度

题目：

问题 J: DS树--二叉树高度

时间限制: 1 Sec  内存限制: 128 MB
提交: 512  解决: 327
[提交][状态][讨论版]
题目描述

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

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

输入

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

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

输出

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

样例输入

1
AB0C00D00
样例输出

3

代码块：

#include <iostream>
using namespace std;

class BiTNode
{
private:
    char data;
    BiTNode *lchild, *rchild;
public:
    BiTNode(){}
    ~BiTNode(){}
    friend class BiTree;
};

class BiTree
{
private:
    BiTNode *root;
    void CreateTree(BiTNode *&p);
    void PreOrderTraverse(BiTNode *p, int i, int &j);
public:
    BiTree(){}
    ~BiTree(){}
    void CreateTree();
    void PreOrder();
};

void BiTree::CreateTree()
{
    CreateTree(root);
}

void BiTree::CreateTree(BiTNode *&p)
{
    char ch;
    cin>>ch;
    if(ch!='0')
    {
        p = new BiTNode;
        p->data = ch;
        CreateTree(p->lchild);
        CreateTree(p->rchild);
    }
    else
        p = NULL;
}

void BiTree::PreOrder()
{
    int i = 1;
    int j = 0;
    PreOrderTraverse(root, i, j);
    cout<<j<<endl;
}

void BiTree::PreOrderTraverse(BiTNode *p, int i, int &j)
{
    if(p)
    {
        if(i>j)
            j = i;
        i++;
        PreOrderTraverse(p->lchild, i, j);
        PreOrderTraverse(p->rchild, i, j);
    }
}

int main(void)
{
    int t;
    cin>>t;
    while(t--)
    {
        BiTree myTree;
        myTree.CreateTree();
        myTree.PreOrder();
    }
    return 0;
}