树的重心dfs

版权声明：若转载请附上原博客链接，谢谢！ https://blog.csdn.net/Link_Ray/article/details/83218063

求法

利用dfs回溯的时候可以算出每个子树的结点个数，从而选出最大的子树，然后与另外的点比较从而求出当前点重心权值。即 $ret = max \{ max \{subtree\}+1, n-max\{subtree\} \}$，+1代表当前结点本身。

点击树的重心模板题

代码

#include <bits/stdc++.h>
using namespace std;
const int maxn = 5e5+5;
vector<int> G[maxn];
int sz[maxn],n,m;
int _min;
int _wei[maxn];
void add(int f,int t) {
	G[f].push_back(t);
}
void dfs(int u, int fa) {
	sz[u] = 1;
	for(int i = 0; i < G[u].size(); ++i) {
		int to = G[u][i];
		if(to != fa) {
			dfs(to,u);
			sz[u] += sz[to];
			_wei[u] = max(_wei[u], sz[to]);
		}
	}
	_wei[u] = max(_wei[u], n-sz[u]);
	_min = min(_min,_wei[u]);
}
int main() {
	_min = 0x3f3f3f3f;
	scanf("%d%d", &n, &m);
	for(int i = 0; i < m; ++i) {
		int u,v;
		scanf("%d%d", &u, &v);
		add(u,v); add(v,u);
	}
	dfs(1,0);
	for(int i = 1; i <= n; ++i)
		if(_wei[i] == _min) {
			cout << i <<" ";
		}
	cout << endl;
	return 0;	
}