比赛 20160229 评测结果 AAAAAEEEEEEEE
题目名称 政党 最终得分 38
用户昵称 张灵犀不和我一般见识真可怕呢(笑 运行时间 0.923 s
代码语言 C++ 内存使用 8.10 MiB
提交时间 2016-02-29 21:49:17
显示代码纯文本
#include<cstdio>
#include<vector>
#include<cmath>
#include<iostream>
using namespace std;
const int SIZEN=40010;
vector<pair<int,int> > s[SIZEN];
vector<int> C[SIZEN];
int dep[SIZEN],dfn[SIZEN*2],dfslist=0,first[SIZEN];
int pos[SIZEN];
int ST[SIZEN*2][20];
int N,M;
int K;
void add(int fr,int to,int lo)
{
	s[fr].push_back(make_pair(to,lo));
	s[to].push_back(make_pair(fr,lo));
}
void dfs(int x,int fa)
{
	dfn[++dfslist]=x;
	first[x]=dfslist;
	for(int i=0;i<s[x].size();i++)
	{
		int v=s[x][i].first,w=s[x][i].second;
		if(v==fa) continue;
		dep[v]=dep[x]+w;
		pos[v]=pos[x]+1;
		dfs(v,x);
		dfn[++dfslist]=x;
	}
}
void make_ST()
{
	int t=int(log2(dfslist+0.0));
	for(int i=1;i<=dfslist;i++) ST[i][0]=dfn[i];
	for(int i=1;i<=t;i++)
	{
		for(int j=1;j<=dfslist;j++)
		{
			if(j+(1<<i)-1>dfslist) break;
			if(pos[ST[j][i-1]]<pos[ST[j+(1<<(i-1))][i-1]]) ST[j][i]=ST[j][i-1];
			else ST[j][i]=ST[j+(1<<(i-1))][i-1];
		}
	}
}
void prepare()
{
	dep[1]=0;pos[1]=0;
	dfs(1,0);
	make_ST();
}
int LCA(int x,int y)
{
	int l=first[x],r=first[y];
	if(l>r) swap(l,r);
	int m=int(log2(r-l+1.0));
	int a=ST[l][m],b=ST[r-(1<<m)+1][m];
	if(pos[a]<pos[b]) return a;
	return b;
}
int main()
{
	freopen("cowpol.in","r",stdin);
	freopen("cowpol.out","w",stdout);
	scanf("%d%d",&N,&K);
	int c,p;
	for(int i=1;i<=N;i++)
	{
		scanf("%d%d",&c,&p);
		C[c].push_back(i);
		if(p!=0)
		add(p,i,1);
	}
	prepare();
	for(int i=1;i<=K;i++)
	{
		int ans=0;
		for(int j=0;j<C[i].size();j++)
		{
			for(int k=0;k<C[i].size();k++)
			{
				if(k==j) continue;
				int a=LCA(C[i][j],C[i][k]);
				ans=max(ans,dep[C[i][j]]+dep[C[i][k]]-2*dep[a]);
			}
		}
		printf("%d\n",ans);
	}
	return 0;
}