CodeForces - 379F - New Year Tree（树的直径）

题目

原始有一棵根为１，有三个叶子２，３，４的树。有ｔ个操作，每次可以在一个叶子下面续上两个节点，每次操作完问当前树的直径。

题解

1.每次加点后，树的直径的端点至多只会变化一个，且必然有一个端点是新加的点。
2. 所以只要分别求他到原直径两端的距离，若大于原直径长度则更新直径。

#include
using namespace std;
int a[1000010][30],d[1000010],n,s,t,mmax;
void init()
{
	memset(a,0,sizeof(a));
	d[1]=0;
	d[2]=d[3]=d[4]=1;
	a[2][0]=a[3][0]=a[4][0]=1;
	n=4;
	s=2;
	t=4;
	mmax=2;
}
void build(int u, int fa)
{
	a[u][0]=fa;
	d[u]=d[fa]+1;
	for(int i=1; i<=21; i++)
		a[u][i]=a[a[u][i-1]][i-1];
}

int sol(int x, int y)
{
	if(x==y) return 0;
	if(d[x]<d[y]) swap(x,y);
	int len=0,i;
	for(i=21; i>=0; i--)
		if(d[x]-(1<<i)>=d[y])
			len+=(1<<i),x=a[x][i];
	for(i=21; i>=0; i--)
		if(a[x][i]!=a[y][i])
			len+=2*(1<<i),x=a[x][i],y=a[y][i];

	if(x!=y) len+=2;
	return len;
}
int main()
{
	int q,x,l;
	init();
	scanf("%d",&q);
	while(q--)
	{
		scanf("%d",&x);
		build(n+1,x);
		build(n+2,x);
		l=sol(n+1,s);
		if(l>mmax)
		{
			t=n+1;
			mmax=l;
		}
		l=sol(n+1,t);
		if(l>mmax)
		{
			s=n+1;
			mmax=l;
		}
		printf("%d\n", mmax);
		n=n+2;
	}
	return 0;
}