樹的重心dfs
阿新 • • 發佈:2018-11-01
求法
利用dfs回溯的時候可以算出每個子樹的結點個數,從而選出最大的子樹,然後與另外的點比較從而求出當前點重心權值。即 ,+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;
}