bzoj5192: [Usaco2018 Feb]New Barns

不想写看zory大佬

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;

int dep[];
int f[][],Bin[];
int LCA(int x,int y)
{
    if(dep[x]<dep[y])swap(x,y);
    for(int i=;i>=;i--)
        if(dep[x]-dep[y]>=Bin[i])x=f[x][i];
    if(x==y)return x;
    for(int i=;i>=;i--)
        if(dep[x]>=Bin[i]&&f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];
    return f[x][];
}
int getdis(int x,int y)
{
    return dep[x]+dep[y]-*dep[LCA(x,y)];
}
int d1[],d2[];
int rt[];

char ss[];
int main()
{
    Bin[]=;for(int i=;i<=;i++)Bin[i]=Bin[i-]*;

    int Q,cnt=,x;
    scanf("%d",&Q);
    while(Q--)
    {
        scanf("%s%d",ss+,&x);
        if(ss[]=='B')
        {
            cnt++;
            if(x==-)
            {
                rt[cnt]=cnt;
                d1[cnt]=d2[cnt]=cnt;
                dep[cnt]=;
            }
            else
            {
                rt[cnt]=rt[x];
                dep[cnt]=dep[x]+;
                f[cnt][]=x;for(int i=;Bin[i]<=dep[cnt];i++)f[cnt][i]=f[f[cnt][i-]][i-];
                int dis=getdis(d1[rt[cnt]],d2[rt[cnt]]);
                if(getdis(d1[rt[cnt]],cnt)>dis)d2[rt[cnt]]=cnt;
                if(getdis(d2[rt[cnt]],cnt)>dis)d1[rt[cnt]]=cnt;
            }
        }
        else printf("%d\n",max( getdis(x,d1[rt[x]]),getdis(x,d2[rt[x]]) ));
    }
    return ;
}