1814 最长链

#include<cstdio>
#include<vector>
using namespace std;
const int N=1e5+100;
int n,dep[N],g[N][21];
int dc[N],tc[N],exde,ans;
short out[N];
vector<int>e[N];
void dfs(int x,int de){
    exde=max(exde,de);
    for(int i=0;i<e[x].size();i++){
        if(!dep[e[x][i]]){
            dep[e[x][i]]=dep[x]+1;
            g[e[x][i]][0]=x;
            dfs(e[x][i],de+1);
        }
    }
}
int lca(int a,int b){
    if(dep[a]<dep[b]) swap(a,b);
    int t=dep[a]-dep[b];
    int ans=0x7fffffff; 
    for(int i=0;i<=20;i++){
        if((1<<i)&t){
            a=g[a][i]; 
        }
    }
    if(a==b) return a;
    for(int i=20;i>=0;i--){
        if(g[a][i]!=g[b][i]){
            a=g[a][i];
            b=g[b][i];  
        }
    }
    return min(g[a][0],g[b][0]);                       
}
int main(){
    scanf("%d",&n);
    if(n==1){printf("1\n");return 0;}
    for(int i=1,x,y;i<=n;i++){
        scanf("%d%d",&x,&y);
        if(x) e[i].push_back(x),out[i]++;
        if(y) e[i].push_back(y),out[i]++;
    }
    for(int i=1;i<=n+1;i++) if(!out[i]) dc[++dc[0]]=i;
    dfs(1,0);
    for(int j=1;j<=20;j++){
        for(int i=1;i<=n;i++){
            g[i][j]=g[g[i][j-1]][j-1];
        }
    }
    int tx=0;
    for(int i=1;i<=dc[0];i++) if(dep[dc[i]]==exde) {tx=dc[i];break;}
    for(int i=1;i<tx;i++){
        int zj=lca(dc[i],tx);
        ans=max(ans,dep[dc[i]]-dep[zj]+dep[tx]-dep[zj]);
    }
    for(int i=tx+1;i<=dc[0];i++){
        int zj=lca(dc[i],tx);
        ans=max(ans,dep[dc[i]]-dep[zj]+dep[tx]-dep[zj]);
    }
    printf("%d\n",ans);
    return 0;
}