AcWing 846. 树的重心

题目链接：

活动 - AcWing

分析：

1. 重心定义：重心是指树中的一个结点，如果将这个点删除后，剩余各个连通块中，点数的最大值最小，那么这个节点被称为树的重心。

2. dfs返回值的含义：返回以删除点为u时，子树连通块和剩余连通块中的最大值。然后ans取所有最大值中的最小值。

3. 因为是无向图，所以用任意点作为起点，其余的点一定能被搜到。（树是无环无向图）

代码：

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 1e5 + 10, M = 2 * N;

int n;
int ans = N;
int h[N], e[M], ne[M], idx;
bool st[N];

void add(int a, int b)
{
	e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}

int dfs(int u)
{
	st[u] = true;
	
	int cnt = 0, sum = 0;
	for (int i = h[u]; i != -1; i = ne[i])
	{
		int j = e[i];
		if (st[j]) continue;
		
		int s = dfs(j);
		cnt = max(cnt, s);
		sum += s;
	}
	
	cnt = max(cnt, n - sum - 1);
	ans = min(ans, cnt);
	
	return sum + 1;
}

int main()
{
	memset(h, -1, sizeof(h));
	
	scanf("%d", &n);
	for (int i = 0; i < n - 1; i ++ )
	{
		int a, b;
		scanf("%d%d", &a, &b);
		add(a, b), add(b, a);
	}
	
	dfs(1);
	
	printf("%d\n", ans);
	
    return 0;
}