POJ3107 Godfather树的重心

POJ3107 Godfather

树的重心模板题

讲解在注释里，树的重心还是比较好理解的

#include<cstdio>
#include<iostream>
using namespace std;
const int N=1e5+3;
int vec[N],siz;
//这道题好像卡vector,自行模拟vector

int nxt[N],head[N],edge[N],tot;
void add(int x,int y){
	nxt[++tot]=head[x];
	head[x]=tot;
	edge[tot]=y;
}
//链式前向星建图

int n,minn=N;
int dp[N],son[N];
//从树中删除该节点后形成的多个树节点尽可能小，
//符合树的重心概念,所以是数的重心。
//dp[i]：去掉i，剩余的最大的树的节点数。
//son[i]：以i为根的树的节点数
void dfs(int me){
	son[me]=1;//带上i自己
	for(int i=head[me];i;i=nxt[i]){
		int v=edge[i];
		if(son[v])continue;
		dfs(v);
		son[me]+=son[v];
		dp[me]=max(dp[me],son[v]);
	}
	dp[me]=max(dp[me],n-son[me]);
	minn=min(dp[me],minn);//找出来最小的dp[i]
	//dad的那棵树大还是me的大？
}
int main(){
	scanf("%d",&n);//这道题还卡cin???我用cin超时？！
	for(int i=1,l,r;i<n;i++){
		scanf("%d%d",&l,&r);
		add(l,r),add(r,l);
	}
	dfs(1);//建的是无向树，哪个节点开始都无所谓。
	for(int i=1;i<=n;i++)if(dp[i]==minn)vec[++siz]=i;
	for(int i=1;i<=siz;i++)printf("%d ",vec[i]);
}