思路:
直接贪心+lca
c o d e code code
#include<iostream>
#include<cstdio>
using namespace std;
int n, m, ans;
int f[1001000][25];
int dep[1000100];
int lca(int x, int y)
{
if(dep[x]<dep[y])
swap(x, y);
int j=20, t=1<<j, k=dep[x]-dep[y];
while(k!=0)
{
if(k>=t)
{
k-=t;
x=f[x][j];
}
j--, t=1<<j;
}
if(x==y)
return x;
for(int j=20; j>=0; j--)
if(f[x][j]!=f[y][j])
x=f[x][j], y=f[y][j];
return f[x][0];
}
int main()
{
f[1][0]=0,f[2][0]=1,f[3][0]=1,f[4][0]=1;
dep[1]=0, dep[2]=1, dep[3]=1, dep[4]=1;
n=4, ans=2;
int s=2, t=3;
scanf("%d", &m);
while(m--)
{
int x;
scanf("%d", &x);
dep[n+1]=dep[x]+1, f[++n][0]=x;
for(int j=1; j<=20; j++)
f[n][j]=f[f[n][j-1]][j-1];
int k=lca(s, t);
int sum=dep[s]+dep[t]-dep[k]*2;
if(sum>ans)
ans=sum;
k=lca(s, n);
sum=dep[s]+dep[n]-dep[k]*2;
if(sum>ans)
ans=sum, t=n;
k=lca(n, t);
sum=dep[n]+dep[t]-dep[k]*2;
if(sum>ans)
ans=sum, s=n;
dep[n+1]=dep[x]+1, f[++n][0]=x;
for(int j=1; j<=20; j++)
f[n][j]=f[f[n][j-1]][j-1];
k=lca(s, t);
sum=dep[s]+dep[t]-dep[k]*2;
if(sum>ans)
ans=sum;
k=lca(s, n);
sum=dep[s]+dep[n]-dep[k]*2;
if(sum>ans)
ans=sum, t=n;
k=lca(n, t);
sum=dep[n]+dep[t]-dep[k]*2;
if(sum>ans)
ans=sum, s=n;
printf("%d\n", ans);
}
return 0;
}