记录编号 442118 评测结果 AAAAAAAAAAAAA
题目名称 [USACO Hol10] 政党 最终得分 100
用户昵称 GravatarFoolMike 是否通过 通过
代码语言 C++ 运行时间 0.625 s
提交时间 2017-08-26 10:36:17 内存使用 8.70 MiB
显示代码纯文本
#include<cstdio>
#include<algorithm>
#include<queue>
using namespace std;
const int N=2e5+10;
int n,m,tp[N],fa[N],size[N],big[N],dep[N];
vector<int> son[N];
void dfs(int x){
	size[x]=1;
	for (int i=0;i<son[x].size();i++){
		int v=son[x][i];
		dep[v]=dep[x]+1;
		dfs(v);
		size[x]+=size[v];
		if (size[big[x]]<size[v]) big[x]=v;
	}
}
int Max[N],ans[N];
vector<int> *q[N];
//dsu on tree
void solve(int x,bool keep){
	for (int i=0;i<son[x].size();i++){
		int v=son[x][i];
		if (v!=big[x]) solve(v,0);
	}
	if (big[x]) solve(big[x],1),q[x]=q[big[x]];
		else q[x]=new vector<int>;
	ans[tp[x]]=max(ans[tp[x]],Max[tp[x]]-dep[x]);
	q[x]->push_back(x);
	Max[tp[x]]=max(Max[tp[x]],dep[x]);
	for (int i=0;i<son[x].size();i++){
		int v=son[x][i];
		if (v==big[x]) continue;
		for (int i=0;i<q[v]->size();i++){
			int u=(*q[v])[i];
			ans[tp[u]]=max(ans[tp[u]],Max[tp[u]]+dep[u]-2*dep[x]);
		}
		for (int i=0;i<q[v]->size();i++){
			int u=(*q[v])[i];
			q[x]->push_back(u);
			Max[tp[u]]=max(Max[tp[u]],dep[u]);
		}
	}
	if (!keep) for (int i=0;i<q[x]->size();i++) Max[tp[(*q[x])[i]]]=-1e9;
}
int main()
{
	freopen("cowpol.in","r",stdin);
	freopen("cowpol.out","w",stdout);
	int __size__ = 32 << 20;
	char *__p__ = (char*)malloc(__size__) + __size__;
	__asm__("movl %0, %%esp\n" :: "r"(__p__));
	scanf("%d%d",&n,&m);
	int root;
	for (int i=1;i<=n;i++){
		scanf("%d%d",&tp[i],&fa[i]);
		if (fa[i]) son[fa[i]].push_back(i);else root=i;
	}
	for (int i=1;i<=m;i++) Max[i]=-1e9;
	dfs(root);solve(root,0);
	for (int i=1;i<=m;i++) printf("%d\n",ans[i]);
	return 0;
}