树的重心题解

Problem F 树的重心

题目描述

​ 对于一棵$n$个结点的无根树，找到一个点，使得把树变成以该点为根的有根树时，最大子树的结点数最小。 换句话说，删除这个点后最大连通块（一定是树）的结点数最小。

输入格式

​ 第一行一个整数$n$, 表示结点数。接下来$n-1$行, 每行两个整数$a$, $b$, 表示结点$a$和$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$节点，各部分分别为：

• 以子节点$j_1$为根的子树 （$绿色区域$）
• 以子节点$j_2$为根的子树 （$蓝色区域$）
• 以子节点$j_3$为根的子树 （$黄色区域$）
• 除 以$u$为根节点的子树 以外的部分（$红色区域$）

变量的规定

sum			指定根节点后，所有当前节点以下的节点总数,在图中是(绿色 + 蓝色 + 黄色)
size		除 以u为根节点的子树 以外的部分(图中红色)
ans			记录当前最优记录答案
ind			记录对应编号
max_part	上述四色区域中，节点最多的。

动态转移方程

因此，$max\_part=max(size,n-sum-1)$

注意！上述的size是没有更新的，也就是上一次运算的结果。本次运算所产生的size就是max_part

$\therefore ans=min(ans,max\_part),即$
$$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;
}

完结，撒花