[gmoj 3555]【GDKOI2014模拟】树的直径

思路：
这道题问树上任意两点距离，我们很自然地想到LCA求距离。
树的直径有一个性质：
现在有两棵树，如果把它们随意连一条边，会变成一棵树，新树的直径的端点一定是之前两棵树的直径的共4个端点的两个。
即我们设(x1,x2)表示一开始那棵树的直径两端，（y1,y2）表示新增两个叶子节点后的直径两端。
最后答案只有在x1~y1, x2_y1,x1x2三条距离取一个最长的，即是新增点后的书的直径。
（因为y1,y2的深度是一样的，所以我们只用随便枚举一点作为树直径的一端）

#include <cstdio>
#include <iostream>

using namespace std;

const int N = 1e5 + 10;
int m, x, f[N << 1][21], som[N << 1], deep[N << 1], ans, num = 4, s = 2, t =4;

int LCA(int x, int y)
{
	if(deep[x] < deep[y]) swap(x, y);
	for(int i = 20; i >= 0; i--)
	 if(deep[f[x][i]] >= deep[y])
	 	x = f[x][i];
	if(x == y) return x;
	for(int i = 20; i >= 0; i--)
	 if(f[x][i] ^ f[y][i])
	  x = f[x][i], y = f[y][i];
	return f[x][0];
}

int Dis(int x, int y)
{
	int res = LCA(x, y);
	return deep[x] + deep[y] - (deep[res] << 1);
}

void add(int now)
{
	deep[++num] = deep[now] + 1;
	f[num][0] = now;
	for(int i = 1; i  <= 20; i++)
	 f[num][i] = f[f[num][i - 1]][i - 1];
}

int main()
{
	f[2][0] = f[3][0] = f[4][0] = 1;
	deep[2] = deep[3] = deep[4] = 1;
//	deep[1] = 1;
	scanf("%d", &m);
	for(int i = 1; i <= m; i++)
	{
		scanf("%d", &x);
		add(x), add(x);
		int dis = Dis(s, t), dis2, dis3;
		dis2 = Dis(s, num);
		dis3 = Dis(t, num);
		if(dis3 >= dis && dis3 >= dis2)
		 s = num, printf("%d\n", dis3), dis = dis3;
		else if(dis2 >= dis && dis2 >= dis3)
		 t = num, printf("%d\n", dis2), dis = dis2;		
		else printf("%d\n", dis); 
	}
	return 0;
}