codeforces379F New Year Tree

Posted 2020-07-06 YJY_

距离一个点最远的点一定是直径的一个端点。
考虑运用这个原理，每次维护一下直径端点即可。

 

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

#define N 500010
#define M 20

int dep[N<<2],f[M][N<<2];

int n;
int q,k;
int v=1,m=2;

int main()
{
	dep[1]=dep[2]=dep[3]=1;
	n=4;
	scanf("%d",&q);
	while (q--)
	{
		scanf("%d",&k);
		k--;
		f[0][n]=f[0][n+1]=k;
		dep[n]=dep[n+1]=dep[k]+1;
		for (int i=1;i<M && f[i-1][f[i-1][n]];i++)
			f[i][n]=f[i][n+1]=f[i-1][f[i-1][n]];
		if (dep[n]>dep[v])
			v=n,++m;
		else if (dep[n]+dep[v]>m)
		{
			int x=v,y=n;
			for (int i=0;i<M && dep[x]!=dep[y];i++)
				if ((dep[x]-dep[y]) & (1<<i))
					x=f[i][x];
			for (int i=M-1;i>=0;i--)
				if (f[i][x]!=f[i][y])
					x=f[i][x],y=f[i][y];
			if (dep[n]+dep[v]-(dep[f[0][x]]<<1)>m)
				++m;
		}
		n+=2;
		printf("%d\n",m);
	}
	return 0;
}