二叉树问题

题目描述
如下图所示的一棵二叉树的深度、宽度及结点间距离分别为：

深度：4 宽度：4（同一层最多结点个数）

结点间距离： ⑧→⑥为8 (3×2+2=8)

⑥→⑦为3 （1×2+1=3）

注：结点间距离的定义：由结点向根方向（上行方向）时的边数×2，

与由根向叶结点方向（下行方向）时的边数之和。

输入格式
输入文件第一行为一个整数n(1≤n≤100)，表示二叉树结点个数。接下来的n-1行，表示从结点x到结点y（约定根结点为1），最后一行两个整数u、v，表示求从结点u到结点v的距离。

输出格式
三个数，每个数占一行，依次表示给定二叉树的深度、宽度及结点u到结点v间距离。

输入输出样例
输入 #1 复制
10
1 2
1 3
2 4
2 5
3 6
3 7
5 8
5 9
6 10
8 6
输出 #1 复制
4
4
8

#include<cstdio>
#include<cmath>
#include<iostream>
using namespace std;
int next[100040];
int to[100040];
int st[1000040];
int de[100040];
int dn[100040];
int ru[100040];
int fa[100040][50];
int n,k,tot=0;
void add(int x,int y)
{
    to[++tot]=y;
    next[tot]=st[x];
    st[x]=tot;
}
void build(int num,int fath)
{
    de[num]=de[fath]+1;
    dn[de[num]-1]++;
    fa[num][0]=fath;
    for(int i=1;i<=29;i++)
    fa[num][i]=fa[fa[num][i-1]][i-1];
    for(int i=st[num];i;i=next[i])
    if(to[i]!=fath) build(to[i],num);
}
int LCA(int x,int y)
{
    if(de[x]<de[y])
    {
        int u=x;
        x=y;
        y=u;
    }
    for(int i=30;i>=0;i--)
    {
        if(de[fa[x][i]]>=de[y]) x=fa[x][i];
        if(x==y) return x;
    }
    for(int i=30;i>=0;i--)
    if(fa[x][i]!=fa[y][i])
    {
        x=fa[x][i];
        y=fa[y][i];
    }
    return fa[x][0];
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n-1;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(y,x);
        add(x,y);
    }
    build(1,0);
    int ma=0;
    for(int i=1;i<=n;i++)
    if(de[i]>ma)
    ma=de[i];
    printf("%d\n",ma);
    int lo=0;
    for(int i=1;i<=ma;i++)
    if(dn[i]>lo)
    lo=dn[i];
    printf("%d\n",lo);
    int x,y;
    scanf("%d%d",&x,&y);
    printf("%d\n",2*(de[x]-de[LCA(x,y)])+de[y]-de[LCA(x,y)]);
    return 0;
}