5433: 数据结构实验：二叉树的直径

5433: 数据结构实验：二叉树的直径

时间限制(普通/Java):1000MS/3000MS     内存限制:65536KByte
总提交:60           测试通过:21

描述

求一棵二叉树的直径，即任意两个结点之间的路径长度最大值，路径可以经过根结点，父子节点之间的路径长度定义为1。

二叉树节点定义如下：

struct TreeNode {

    int val;

    struct TreeNode *left;

    struct TreeNode *right;

};

题目部分代码已经完成，您只需要补充并提交以下函数：

int diameterOfBinaryTree(struct TreeNode* root);

输入

输入为若干个整数（不超过512）表示一棵二叉树顺序表示时的结点元素值，其中0表示二叉树对应结点为空。输入以-1结束。 

输出

输出二叉树的直径。

样例输入

1 2 3 4 5 -1

样例输出

3

题目来源

TOJ

#include 
#include 

struct TreeNode {
    int val;
    struct TreeNode *left,*right;
};

struct TreeNode a[520];

struct TreeNode* Creat(struct TreeNode* root)
{
    int k=0,i,j,n;
    while(scanf("%d",&n),n!=-1)
        a[k++].val=n;
    for(i=0,j=1;i=k||a[i+j].val==0) a[i].left=NULL;
        else a[i].left=&a[i+j];
        if(i+j+1>=k||a[i+j+1].val==0) a[i].right=NULL;
        else a[i].right=&a[i+j+1];
    }
    return &a[0];
}

int maxn=0;

int depth(struct TreeNode* root)
{
    if(!root) return 0;
    int ls=depth(root->left);
    int rs=depth(root->right);
    return ls>rs?(ls+1):(rs+1);
}

void dfs(struct TreeNode* root)
{
    if(root)
    {
        int n=depth(root->left),m=depth(root->right);
        if(n+m>maxn) maxn=n+m;
        dfs(root->left);
        dfs(root->right);
    }
}

int diameterOfBinaryTree(struct TreeNode* root)
{
    dfs(root);
    return maxn;
}

int main()
{
    struct TreeNode* root=NULL;
    root=Creat(root);
    printf("%d\n",diameterOfBinaryTree(root));
    return 0;
}