NKOI 2044 奶牛政坛

【RMQ&LCA】奶牛政坛

Time Limit:10000MS  Memory Limit:65536K
Total Submit:22 Accepted:16
Case Time Limit:1000MS

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 

Hint

数据弱，无需手工栈

Source

Usaco2010 Hol

这题相对来说就比较水了，根据题意我们可以知道每条边的长度为1，我们先用DFS把树建好之后就可以知道每条边的depth值了，根据树的直径的性质，先分别找出每一个政党中深度最大的一个编号，然后找出离这个编号最远的一个同政党的编号的距离的最大值就可以了

这里求点的距离可以用depth计算，ans=max(ans,deep[a]+deep[b]-2*deep[h])其中h为最近公共祖先

#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;

const int maxn=200005;
int n,depth[maxn],f[maxn][21],s,a,b,p[maxn],m,tot;
int END[maxn<<1],NEXT[maxn<<1],last[maxn],point[maxn];
int ans[maxn];
bool mark[maxn];
queue<int>q;

inline void _read(int &x){
	char t=getchar();bool sign=true;
	while(t<'0'||t>'9')
	{if(t=='-')sign=false;t=getchar();}
	for(x=0;t>='0'&&t<='9';t=getchar())x=x*10+t-'0';
	if(!sign)x=-x;
}

void insert(int a,int b){
	END[++tot]=b;
	NEXT[tot]=last[a];
	last[a]=tot;
}

void dfs(int v,int dp){
	depth[v]=dp,mark[v]=1;
	int i,j,k=ceil(log(depth[v])/log(2));
	for(i=1;i<=k;i++)
		f[v][i]=f[f[v][i-1]][i-1];
	for(i=last[v],j=END[i];i;i=NEXT[i],j=END[i])
		if(!mark[j]){
			f[j][0]=v;
			dfs(j,dp+1);
		}
}

int LCA(int x,int y)
{
	int i,k;
	if(depth[x]<depth[y])swap(x,y);
	k=depth[x]-depth[y];
	for(i=0;i<=s;i++)
	    if(k&(1<<i))
	    	x=f[x][i];
	if(x==y)return x;
	k=ceil(log(depth[x])/log(2));
	for(i=k;i>=0;i--)
	    if(f[x][i]!=f[y][i])
		    x=f[x][i],y=f[y][i];
	return f[x][0];
}

int main(){
	_read(n);_read(m);
	int i,j,root;
	for(i=1;i<=n;i++){
		_read(p[i]);_read(b);
		if(b==0){root=i;continue;}
		insert(b,i);
	}
	dfs(root,1);
	for(i=1;i<=n;i++)
	    if(depth[i]>depth[point[p[i]]])
	        point[p[i]]=i;//找出深度最深的一个点
	s=ceil(log(n)/log(2));
	for(i=1;i<=n;i++){
		int st=LCA(i,point[p[i]]);
		ans[p[i]]=max(ans[p[i]],depth[i]+depth[point[p[i]]]-depth[st]*2);
	}
	for(i=1;i<=m;i++)
	    printf("%d\n",ans[i]);
}