比赛 防止浮躁的小练习V0.1 评测结果 AAAAAAAAAAAAA
题目名称 政党 最终得分 100
用户昵称 NVIDIA 运行时间 4.375 s
代码语言 C++ 内存使用 19.48 MiB
提交时间 2016-10-07 18:42:54
显示代码纯文本
#include <algorithm>
#include <vector>
#include <iostream>
#include <queue>
#include <cstdio>
#define N 200010
#define M 20
using namespace std;
int n,K;
vector<int> G[N];
vector<int> F[N];
int deep[N]={0};
int p[N][M]={0};
bool l[N]={0};
int o;
void BFS(int s)
{
	int i,u,v;
	queue<int> q;
	for(i=1;i<=n;i++)l[i]=0;
	q.push(s);
	while(!q.empty())
	{
		u=q.front();
		q.pop();
		l[u]=1;
		for(i=0;i<G[u].size();i++)
		{
			v=G[u][i];
			if(!l[v])
			{
				p[v][0]=u;
				deep[v]=deep[u]+1;
				q.push(v);
			}
		}
	}
}
void LCB(int u)
{
	int i,v;
	l[u]=1;
	for(i=1;i<M;i++)
	{
		p[u][i]=p[p[u][i-1]][i-1];
		if(!p[u][i])break;
	}
	for(i=0;i<G[u].size();i++)
	{
		v=G[u][i];
		if(!l[v])LCB(v);
	}
}
int LCA(int u,int v)
{
	int i,d,t;
	if(deep[u]<deep[v])
	{
		t=u;
		u=v;
		v=t;
	}
	d=deep[u]-deep[v];
	for(i=0;i<M;i++)if((1<<i)&d)u=p[u][i];
	if(u==v)return u;
	for(i=M-1;i>=0;i--)
	{
		if(p[u][i]!=p[v][i])
		{
			u=p[u][i];
			v=p[v][i];
		}
	}
	u=p[u][0];
	return u;
}
int dis(int u,int v)
{
	int fa,ans=0;
	fa=LCA(u,v);
	ans=deep[u]+deep[v]-2*deep[fa];
	return ans;
}
void read()
{
	int i,j,u,v,w;
	cin>>n>>K;
	for(i=1;i<=n;i++)
	{
		u=i;
		cin>>w>>v;
	    F[w].push_back(i);
		if(v==0)continue;
		G[u].push_back(v);
		G[v].push_back(u);
	}
	BFS(1);
	for(i=1;i<=n;i++)l[i]=0;
	LCB(1);
}
void make(int s)
{
	int i,j,u,v;
	int temp=0,best=0,st=0,en=0;
	u=F[s][0];
	for(i=0;i<F[s].size();i++)
	{
		v=F[s][i];
		temp=dis(u,v);
		if(temp>best)
		{
			best=temp;
			st=v;
		}
	}
	best=0;
	for(i=0;i<F[s].size();i++)
	{
		v=F[s][i];
		temp=dis(st,v);
		if(temp>best)
		{
			best=temp;
			en=v;
		}
	}
	cout<<best<<endl;
}
void work()
{
	int i,j;
	for(i=1;i<=K;i++)make(i);
}
int main()
{
	freopen("cowpol.in","r",stdin);
	freopen("cowpol.out","w",stdout);
	read();
	work();
	return 0;
}