c++树的重心

• 树的重心是什么

在一棵树上，找到一个点使其最大子树的节点数最少，则这个点就是树的重心，树的重心可能不只一个。

如图，2是树的重心。

• 树的重心的求法

1.根据定义来求：

求出每个节点的子树节点数的最大值，再求出其中的最小值，输出子树节点数的最大值为这些节点中最小的的那个节点。

1. 先假设此节点为树的重心，再进行遍历，如果发现这个节点的某个子树的节点数超过了总节点数的一半，则这个节点就不是树的重心。

三．代码实现

方法一

#include<bits/stdc++.h>
#define int long long
using namespace std;
vector<int> v[110000];//树
int n, son[110000], ans[110000];
//n是总结点数,son是该子节的节点数(包括该点本身),ans是该节点的子树节点数中的最大值
void dfs(int x, int fa) {
	son[x] = 1;
	for (int i = 0; i < v[x].size(); i++)
		if (v[x][i] != fa) {
			dfs(v[x][i], x);
			son[x] += son[v[x][i]]; //求出该子节的节点数(包括该点本身)
			ans[x] = max(ans[x], son[v[x][i]]); //求出该节点的子树节点数中的最大值
		}
	ans[x] = max(ans[x], n - son[x]); //不要忘记它上方的节点数
}
signed main() {
	cin >> n;
	for (int i = 1; i < n; i++) {
		int a, b;
		cin >> a >> b;
		v[a].push_back(b);
		v[b].push_back(a);
	}//建树
	dfs(1, 0);
	int mn = INT_MAX;
	for (int i = 1; i <= n; i++)
		mn = min(mn, ans[i]); //求出子树节点数中的最大值中的最小值
	for (int i = 1; i <= n; i++)
		if (ans[i] == mn)
			cout << i << endl;
	return 0;
}

方法二

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N = 50000 + 10;
vector<int>s[N];//树
int n, son[N]; //n是总结点数,son是该子节的节点数(包括该点本身)
bool vis[N];//vis代表该节点是不是重心
void dfs(int u, int fa) {
	vis[u] = true; //假设它是重心
	son[u] = 1;
	for (int i = 0; i < s[u].size(); i++)
		if (s[u][i] != fa) {
			dfs(s[u][i], u);
			son[u] += son[s[u][i]]; //求出该子节的节点数(包括该点本身)
			if (son[s[u][i]] > n / 2)
				vis[u] = false; //判断
		}
	if (n - son[u] > n / 2)
		vis[u] = false; //不要忘记它上方的节点数
}
signed main() {
	cin >> n;
	for (int i = 1; i < n; i++) {
		int x, y;
		cin >> x >> y;
		s[x].push_back(y);
		s[y].push_back(x);
	}//建树
	dfs(1, 0);
	for (int i = 1; i <= n; i++)
		if (vis[i])
			cout << i << endl;
}