比赛 防止浮躁的小练习V0.1 评测结果 AAAAAAAAAAAAA
题目名称 政党 最终得分 100
用户昵称 浮生随想 运行时间 2.294 s
代码语言 C++ 内存使用 47.62 MiB
提交时间 2016-10-07 16:46:37
显示代码纯文本
#include<cstdlib>
#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<cmath>
using namespace std;
const int maxn=400010;
int a[maxn],p[maxn],n,m,root,len=0,head[maxn],st,ed,cnt=0;
int id[maxn],f[maxn][20],l;
vector<int>be[maxn];
struct edge{
	int to,next;
}e[maxn<<1];
void dfs(int x){
	for(int i=head[x];i;i=e[i].next){
		int to=e[i].to;
		if(!id[to]){
			f[to][0]=x;
			id[to]=id[x]+1;
			dfs(to);
		}
	}
}
int lca(int x,int y){
	
	if(id[x]<id[y])swap(x,y);
	for(int j=l-1;j>=0;j--)
		if(id[f[x][j]]>=id[y])
			x=f[x][j];
	if(x==y)	return x;
	for(int j=l-1;j>=0;j--)
		if(f[x][j]!=f[y][j]){
			x=f[x][j];y=f[y][j];
		}
	return f[x][0];
}
void insert(int x,int y){
	len++;e[len].to=y;e[len].next=head[x];head[x]=len;
	//cout<<x<<" "<<y<<endl;
}
int ST(int x){
	int rt=be[x][0];//cout<<"hehe"<<endl;
	int nn=be[x].size(),cnt,ans=0;
	for(int i=0;i<nn;++i){
		int y=be[x][i],k=lca(rt,y),d=id[rt]+id[y]-id[k]*2;
		if(ans<d)	ans=d,cnt=y;
	}
	for(int i=0;i<nn;++i){
		int y=be[x][i],k=lca(cnt,y),d=id[cnt]+id[y]-id[k]*2;
		if(ans<d)	ans=d;
	}
	return ans;	
}
int main(){
	freopen("cowpol.in","r",stdin);
	freopen("cowpol.out","w",stdout);
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;++i){
		int y,x;
		scanf("%d%d",&y,&x);
		be[y].push_back(i);
		if(!x){root=i;continue;}
		insert(x,i);
		insert(i,x);
	}
	l=(int)(log(n*1.0)/log(2.0))+1;
	f[root][0]=root;
	id[root]=1;
	dfs(root);
	for(int j=1;j<l;++j)
		for(int i=1;i<=n;++i)	f[i][j]=f[f[i][j-1]][j-1];	
	for(int i=1;i<=m;++i)	printf("%d\n",ST(i));
	//system("pause");
	return 0;
}