Problem F 树的重心
题目描述
对于一棵 n n n个结点的无根树,找到一个点,使得把树变成以该点为根的有根树时,最大子树的结点数最小。 换句话说,删除这个点后最大连通块(一定是树)的结点数最小。
输入格式
第一行一个整数 n n n, 表示结点数。接下来 n − 1 n-1 n−1行, 每行两个整数 a a a, b b b, 表示结点 a a a和 b b b有边。
输出格式
一行,两个整数,第一整数为删除的结点编号,第二个整数为删除此结点后其最大子树的结点数量。
样例
输入
11
3 4
1 2
3 1
5 3
5 10
4 7
11 6
4 8
5 9
3 6
输出
3 3
讲解
对于无根树,我们不必指定每一个节点为根在求最大子树的节点数,只需要指定一个元素为根,再遍历节点,看节点各子树的节点数 以及 除了以该节点为根的子树中其它部分的节点数。
看下图:
对于此时的 u u u节点,各部分分别为:
- 以子节点 j 1 j_1 j1为根的子树 ( 绿 色 区 域 \color{green} 绿色区域 绿色区域)
- 以子节点 j 2 j_2 j2为根的子树 ( 蓝 色 区 域 \color{blue} 蓝色区域 蓝色区域)
- 以子节点 j 3 j_3 j3为根的子树 ( 黄 色 区 域 \color{yellow}黄色区域 黄色区域)
- 除 以 u u u为根节点的子树 以外的部分( 红 色 区 域 \color{red}红色区域 红色区域)
变量的规定
sum 指定根节点后,所有当前节点以下的节点总数,在图中是(绿色 + 蓝色 + 黄色)
size 除 以u为根节点的子树 以外的部分(图中红色)
ans 记录当前最优记录答案
ind 记录对应编号
max_part 上述四色区域中,节点最多的。
动态转移方程
因此, m a x _ p a r t = m a x ( s i z e , n − s u m − 1 ) max\_part = max(size, n - sum - 1) max_part=max(size,n−sum−1)
注意!上述的size是没有更新的,也就是上一次运算的结果。本次运算所产生的size就是max_part
∴
a
n
s
=
m
i
n
(
a
n
s
,
m
a
x
_
p
a
r
t
)
,
即
\therefore ans = min(ans, max\_part),即
∴ans=min(ans,max_part),即
a
n
s
=
m
i
n
(
a
n
s
,
m
a
x
(
s
i
z
e
,
n
−
s
u
m
−
1
)
)
ans = min(~ans,~\color{blue}{max(size,~n - sum - 1)}\color{black}~)
ans=min( ans, max(size, n−sum−1) )
传送门
源代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <bitset>
using namespace std;
const int N = 1000003, M = N << 1;
int n;
int e[M], nxt[M], h[N], cnt;
int ans = 0x3f3f3f3f;
int ind;
bitset <M> vis;
void add(int u, int v)
{
e[cnt] = v;
nxt[cnt] = h[u];
h[u] = cnt++;
}
int dfs(int u)
{
vis[u] = 1;
int size = 0, sum = 0;
for(int i = h[u]; ~i; i = nxt[i])
{
int j = e[i];
if(!vis[j])
{
int t = dfs(j);
size = max(size, t);
sum += t;
}
}
size = max(size, n - sum - 1);
if(size <= ans)
{
ans = size;
ind = u;
}
return sum + 1;
}
int main()
{
memset(h, -1, sizeof(h));
scanf("%d", &n);
for(int i = 1; i < n; ++i)
{
int a, b;
scanf("%d%d", &a, &b);
add(a, b);
add(b, a);
}
dfs(1);
printf("%d %d", ind, ans);
return 0;
}
完结,撒花