2021.07.14【NOIP提高B组】模拟 树的直径

思路：

直接贪心+lca

$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;
}