P3884 [JLOI2009]二叉树问题

#include<iostream>
#include<cstdio>
using namespace std;
int head[2*500001];
int p;
int md;
int mk;
int wei[10000];
struct b{
    int to;
    int ne;
} e[2*500001];
int fa[2*500001][30];
int dep[2*500001];
int n,m,s;
int logg[2*500001];
int x,y;
void dfs(int now,int fat){
    dep[now]=dep[fat]+1;
    md=max(md,dep[now]);
    wei[dep[now]]++;
    mk=max(mk,wei[dep[now]]);
    fa[now][0]=fat;
    for(int i=1;(1<<i)<=dep[now];++i){
        fa[now][i]=fa[fa[now][i-1]][i-1];
    }
    for(int i=head[now];i;i=e[i].ne){
        if(e[i].to!=fat)
        dfs(e[i].to,now);
    }
}
int lca(int x,int y){
    int ans=0;
    if(dep[x]>dep[y]){
        int v=dep[y];
    ans+=(2*(dep[x]-dep[y]));
    while(dep[x]>dep[y]){
    x=fa[x][logg[dep[x]-dep[y]]-1];
    }
    if(x==y)
    return ans;
    for(int k=logg[dep[x]]-1;k>=0;k--){
        if(fa[x][k]!=fa[y][k]){
            x=fa[x][k];
            y=fa[y][k];
        }
    }
    ans+=(3*(v-dep[fa[x][0]]));
    return ans;
    }
    else{
        ans+=(dep[y]-dep[x]);
        int v=dep[x];
    while(dep[x]<dep[y]){
    y=fa[y][logg[dep[y]-dep[x]]-1];
    }
    if(x==y)
    return ans;
    for(int k=logg[dep[x]]-1;k>=0;k--){
        if(fa[x][k]!=fa[y][k]){
            x=fa[x][k];
            y=fa[y][k];
        }
    }
    ans+=(3*(v-dep[fa[x][0]]));
    return ans;
        
    }
}

void add(int f,int t){
    p++;
    e[p].to=t;
    e[p].ne=head[f];
    head[f]=p;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n-1;++i){
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    dfs(1,0);
    cout<<md<<endl<<mk<<endl;
    for(int i=1;i<=n;++i)
        logg[i]=logg[i-1]+(1<<logg[i-1]==i);
    for(int i=1;i<=1;++i){
        scanf("%d%d",&x,&y);
        cout<<lca(x,y)<<endl;
    }
    return 0;
}