小G有一个大树
题目地址:
基本思路:
树的重心模板题,我们顺便来复习一下重心的性质:
树的重心的定义就是题意:找到一个点,其所有的子树中最大的子树节点数最少。
我们再来看下百度百科里树的重心的性质:
- 树中所有点到某个点的距离和中,到重心的距离和是最小的,如果有两个重心,他们的距离和一样。
- 把两棵树通过一条边相连,新的树的重心在原来两棵树重心的连线上。
- 一棵树添加或者删除一个节点,树的重心最多只移动一条边的位置。
- 一棵树最多有两个重心,且相邻。
我们再来看树的重心的一般求法:
void dfs(int u,int p) {
sz[u] = 1;
for (int i = head[u]; i != -1; i = edge[i].next) {
int to = edge[i].to;
if (to == p) continue;
dfs1(to, u);
sz[u] += sz[to];//更新u为节点的子树的节点数;
f[u] = max(f[u], sz[to]);//子树最大的节点数;
}
f[u] = max(f[u], n - sz[u]);//n-sz[u]就是把它父节点作为根的子树的节点数;
if (f[u] < mi) {//直接取子树中最大的子树节点数最少的节点为重心;
mi = f[u];
res = u;
}
} 参考代码:
#pragma GCC optimize(2)
#pragma GCC optimize(3)
#include <bits/stdc++.h>
using namespace std;
#define IO std::ios::sync_with_stdio(false); cin.tie(0)
#define int long long
#define SZ(x) ((int)(x).size())
#define all(x) (x).begin(), (x).end()
#define rep(i, l, r) for (int i = l; i <= r; i++)
#define per(i, l, r) for (int i = l; i >= r; i--)
#define mset(s, _) memset(s, _, sizeof(s))
#define pb push_back
#define pii pair <int, int>
#define mp(a, b) make_pair(a, b)
#define INF 0x3f3f3f3f
inline int read() {
int x = 0, neg = 1; char op = getchar();
while (!isdigit(op)) { if (op == '-') neg = -1; op = getchar(); }
while (isdigit(op)) { x = 10 * x + op - '0'; op = getchar(); }
return neg * x;
}
inline void print(int x) {
if (x < 0) { putchar('-'); x = -x; }
if (x >= 10) print(x / 10);
putchar(x % 10 + '0');
}
const int maxn = 1100;
struct Edge{
int to,next;
}edge[maxn << 1];
int n,cnt,head[maxn];
void add_edge(int u,int v){
edge[++cnt].next = head[u];
edge[cnt].to = v;
head[u] = cnt;
}
int ans,sz[maxn],f[maxn],mn = INF;
void dfs(int u,int par) {
sz[u] = 1;
for (int i = head[u]; i != -1; i = edge[i].next) {
int to = edge[i].to;
if (to == par) continue;
dfs(to, u);
sz[u] += sz[to];
f[u] = max(f[u], sz[to]);
}
f[u] = max(f[u], n - sz[u]);
if (f[u] <= mn) {
if (f[u] == mn) ans = min(ans,u);
else ans = u;
mn = f[u];
}
}
signed main() {
IO;
cnt = 0;
mset(head,-1);
cin >> n;
for(int i = 1 ; i < n ; i++){
int u,v;
cin >> u >> v;
add_edge(u,v);
add_edge(v,u);
}
ans = 0;
dfs(1,0);
cout << ans << ' ' << mn << '\n';
return 0;
}
京公网安备 11010502036488号