sgu 134 Centroid 树的重心

题目链接

求树的重心。本质上这题是模拟题。

把一棵树上的一个点u删掉，这棵树会分成几棵小树，这些小树的节点数的最大值就是点u的权值。

权值最小的点是这棵树的重心。

#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <vector>
#include <set>
#include <queue>
#include <map>
using namespace std;
#define INF 1e9
#define maxn 16010
#define rep(i,x,y) for(int i=x;i<=y;i++)
#define mset(x) memset(x,0,sizeof(x))

int n, a, b;
vector<int> G[maxn];
int val[maxn];//点的权值
vector<int> ans;

void init(){
	rep(i,1,n)	G[i].clear();
	mset(val);
	ans.clear();
}

int dfs(int u, int pa){
	int sum=1;//u自己
	for(int i=0; i<G[u].size(); i++){
		int v = G[u][i];
		if(v==pa)	continue;
		int cntv = dfs(v, u);
		sum += cntv;
		val[u] = max(val[u], cntv);
	}
	val[u] = max(val[u], n-sum);
	return sum;
}

int main(){
//	freopen("a.txt","r",stdin);
//	freopen(".out","w",stdout);
	cin>>n;
	init();
	rep(i,1,n-1){
		scanf("%d%d", &a, &b);
		G[a].push_back(b);
		G[b].push_back(a);
	}
	dfs(1, -1);

	int minv = INF;
	rep(i,1,n)	minv = min(val[i], minv);
	int cencnt = 0;
	
	rep(i,1,n)	if(val[i]==minv){ ans.push_back(i); cencnt++; }
	printf("%d %d\n", minv, cencnt);
	for(int i=0; i<ans.size(); i++){
		if(i)	putchar(' ');
		printf("%d", ans[i]);
	}
	puts("");
	return 0;
}