18923 二叉树的直径

Description

给定一棵二叉树，你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。
          1
         / \
        2   3
       / \     
      4   5   
答案为3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

输入格式

共n行。
第一行一个整数n，表示有n个结点，编号为1至n。
第二行至第n行，每行有两个整数x和y，表示在二叉树中x为y的父节点。x第一次出现时y为左孩子

 

输出格式

输出二叉树的直径。

输入样例

5
1 2
1 3
2 4
2 5

输出样例

3

直径长度就是左子树的深度加上右子树的深度

#include<stdio.h>
#include<stdlib.h>
#include<malloc.h>
typedef struct BiTNode
{
    int data;
    struct BiTNode *lchild,*rchild;
}BiTNode,*BiTree;

//递归求二叉树的深度
int Depth(BiTree T)
{
    if(!T)
    {
        return 0;
    }
    else
    {
        int left=Depth(T->lchild);
        int right=Depth(T->rchild);
        return (left>right?left:right)+1;
    }
}

int main()
{
    int n,ans=0;
    scanf("%d",&n);
    BiTree T[n+1];
    for(int i=1;i<=n;i++)
    {
        T[i]=(BiTNode*)malloc(sizeof(BiTNode));
        T[i]->lchild=NULL;
        T[i]->rchild=NULL;
    }
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d %d",&x,&y);
        if(!(T[x]->lchild))
        {
            T[x]->lchild=T[y];
        }
        else
        {
            T[x]->rchild=T[y];
        }
    }
    //因为题目中说了根是从1开始的
    ans=Depth(T[1]->lchild)+Depth(T[1]->rchild);//左子树的深度+右子树的深度
    printf("%d",ans);
    return 0;
}