树的中心
前置知识
树的中心
树的中心的定义,树的中心
暴力算法
树的中心就是直径的中心.
设
- 直径的中心为c
- 任意一点 u ,
表示直径的端点的集合
已知定理1: 树上任意一点u 到达的最远点一定是直径的端点
根据u的位置,分情况讨论:
情况1: 点u在直径上, 那么显然
对于任意的点u, u在直径上, 那么
情况2: 点u不在直径上, 那么显然
先找到u到达直径上的最近点t,
- 如果
, 那么显然$f(u,T) = dis(u,t) + dis( t,e_i) > dis(c,e_i) = f(c,T)$ - 如果
, 那么显然, 那么这个 就是情况1,
所以, 树的中心一定是直径的中心
代码
只要先用两次dfs 找到直径的两个端点, 然后再找到直径的中心即可
cpp
copy
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int n;
vector<int> g[N];
int dis[N];
int maxdis, maxid;
void dfs(int u, int fa) {
if (dis[u] > maxdis) {
maxdis = dis[u];
maxid = u;
}
for (auto v : g[u]) {
if (v == fa) continue;
dis[v] = dis[u] + 1;
dfs(v, u);
}
}
int main() {
cin >> n;
for (int i = 1; i < n; i++) {
int u, v;
cin >> u >> v;
g[u].push_back(v);
g[v].push_back(u);
}
dfs(1, -1);
maxdis = 0;
dfs(maxid, -1);
int cen = maxid;
int len = maxdis;
if (len % 2 == 0) {
for (int i = 1; i <= len / 2; i++) {
for (auto v : g[cen]) {
if (dis[v] == dis[cen] - 1) {
cen = v;
break;
}
}
}
} else {
for (int i = 1; i <= len / 2; i++) {
for (auto v : g[cen]) {
if (dis[v] == dis[cen] - 1) {
cen = v;
break;
}
}
}
}
cout << cen << endl;
return 0;
}