[Usaco2010 Hol]cowpol 奶牛政坛

Description

农夫约翰的奶牛住在N (2 <= N <= 200,000)片不同的草地上，标号为1到N。恰好有N-1条单位长度的双向道路，用各种各样的方法连接这些草地。而且从每片草地出发都可以抵达其他所有草地。也就是说，这些草地和道路构成了一种叫做树的图。输入包含一个详细的草地的集合，详细说明了每个草地的父节点P_i (0 <= P_i <= N)。根节点的P_i == 0, 表示它没有父节点。因为奶牛建立了1到K一共K (1 <= K <= N/2)个政党。每只奶牛都要加入某一个政党，其中， 第i只奶牛属于第A_i (1 <= A_i <= K)个政党。而且每个政党至少有两只奶牛。 这些政党互相吵闹争。每个政党都想知道自己的“范围”有多大。其中，定义一个政党的范围是这个政党离得最远的两只奶牛（沿着双向道路行走）的距离。 比如说，记为政党1包含奶牛1，3和6，政党2包含奶牛2，4和5。这些草地的连接方式如下图所 示（政党1由-n-表示）： 政党1最大的两只奶牛的距离是3（也就是奶牛3和奶牛6的距离）。政党2最大的两只奶牛的距离是2（也就是奶牛2和4，4和5，还有5和2之间的距离）。 帮助奶牛们求出每个政党的范围。

Input

第一行: 两个由空格隔开的整数: N 和 K * 第2到第N+1行: 第i+1行包含两个由空格隔开的整数: A_i和P_i。

Output

第1到第K行: 第i行包含一个单独的整数，表示第i个政党的范围。

Sample Input

6 2
1 3
2 1
1 0
2 1
2 1
1 5

Sample Output

3
2

题目大意：
给你一棵有n个节点的树，在数的节点处有k种不同的标记，现问每种不同的标记两个相差的最远节点距离为多少。

题解：
如果只求一个距离的话，那么随便找一个点，找出一个离它最远的点，再以这个最远的点寻找，再以dfs找出离它最远的点即可。

但是这题数据范围很大，并且寻找的不止一个数据，所以按照原算法肯定不行。那我们不妨优化一下，既然要搜相差最远的两个点，在树上，有且只有一条边，并且每相邻两个点的距离是一定的，那么当前这个标记的其中一个点必定是在这个当前标记内深度最深的一个点（用dfs查找），再以这个点开始搜索即可。

搜索优化：一般使用的是dfs搜索，但在此题内，必定会超时，那么我们可以
LCA（最近公共祖先）由于两个点是一个节点进行分叉，分叉再分叉…这样分叉n次后得到的两个点，那我们就可以用RMQ（倍增思想）来得出这个点，
并以两个节点的深度减去最近公共祖先的深度即可。
RMQ：以当前节点来通过翻倍来以此获得下一个节点的值

代码实现如下：

#include<bits/stdc++.h>
#define N 400001
using namespace std;
int f[N][20],m[N];
int dep[N],a[N],pre[N],now[N],son[N],tot,sum[N];
bool pp[N];
struct k {
	int dep,s;
} maxx[100001];
void addx(int x,int y) {
	pre[++tot]=now[x];
	now[x]=tot;
	son[tot]=y;
}
void dfs(int u,int d) {
	dep[u]=d;
	if(maxx[a[u]].dep<d) {
		maxx[a[u]].dep=d;
		maxx[a[u]].s=u;
	}
	pp[u]=1;
	for(int i=now[u]; i; i=pre[i]) {
		int c=son[i];
		if(!pp[c]) {
			f[c][0]=u;
			dfs(c,d+1);
		}
	}
}
int lca(int x,int y) {
	if(dep[x]<dep[y])
		swap(x,y);
	for(int i=19; i>=0; i--)
		if(dep[f[x][i]]>=dep[y])
			x=f[x][i];
	if(x==y)
		return x;
	for(int i=19; i>=0; i--)
		if(f[x][i]!=f[y][i])
			x=f[x][i],y=f[y][i];
	return f[x][0];
}
int main() {
	int x,y,s,q,n;
	scanf("%d%d",&n,&q);
	for(int i=1; i<=n; i++) {
		scanf("%d%d",&x,&y);
		a[i]=x;
		addx(y,i);
	}
	dfs(0,0);
	memset(pp,false,sizeof(pp));
	for(int i=1; i<=q; i++)
		pp[maxx[a[i]].s]=1;
	for(int j=1; j<20; j++)
		for(int i=1; i<=n; i++)
			f[i][j]=f[f[i][j-1]][j-1];
	for(int i=1; i<=n; i++) {
		int l;
		if(pp[i]==false) {
			l=lca(i,maxx[a[i]].s);
			int ans=dep[i]-dep[l]+maxx[a[i]].dep-dep[l];
			if(sum[a[i]]<ans) {
				sum[a[i]]=ans;
			}
		}
	}
	for(int i=1; i<=q; i++)
		printf("%d\n",sum[i]);
}

第六篇CSDN，可能不是很详细，望谅解，谢谢！