1. 程式人生 > >樹的重心dfs

樹的重心dfs

求法

利用dfs回溯的時候可以算出每個子樹的結點個數,從而選出最大的子樹,然後與另外的點比較從而求出當前點重心權值。即 r e t = m a x {

m a x { s u b t r e e
} + 1 , n m a x { s
u b t r e e } } 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;	
}