C++算法题 # 38 树的重心（树的深度优先遍历)

题目描述
给定一颗树，树中包含 n 个结点（编号 1∼n）和 n−1 条无向边。

请你找到树的重心，并输出将重心删除后，剩余各个连通块中点数的最大值。

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

输入格式
第一行包含整数 n，表示树的结点数。

接下来 n−1 行，每行包含两个整数 a 和 b，表示点 a 和点 b 之间存在一条边。

输出格式
输出一个整数 m，表示将重心删除后，剩余各个连通块中点数的最大值。

数据范围
$1\le n\le 10^5$
输入样例

9
1 2
1 7
1 4
2 8
2 5
4 3
3 9
4 6

输出样例：

4

思路
本题首先就是要对题目进行理解，理解重心的意思。

这里使用深度优先遍历去遍历每一个子树，每个子树也只需要遍历一次即可，我们可以用一个数组来记录那些点已经遍历过了。
这里需要注意一个就是删除一个节点后连通块如何统计的问题，删除某个节点后主要分为2大部分，一个是以该节点为根节点的子节点的连通块，另一部分是该节点的父亲节点的连通块。

然后这里还有个补充，关于数和图的存储方式，算法使用数组模拟速度较快。下面是存储示意图：

代码示例

#include <iostream>
#include <cstring>
using namespace std;

const int N = 1e5 + 10, M = N * 2; 
int n;
int h[N], e[M], ne[M], idx;//这里 M 是 N的两倍  是因为需要存放无向边 即可以来回遍历父子节点
int st[N];
int ans = N;// 这里先设置一个最大值，为了找最小的值 注意不要初始化为0

void add(int a, int b)// 创建节点a到节点b的连接
{
	e[idx] = b, ne[idx] = h[a], h[a] = idx ++;

}

int  dfs(int u)// 功能：得到节点u的子树中的最大点数
{
	st[u] = true; // 标记已被搜寻过的节点
	int sum = 1, res = 0;
	
	for(int i = h[u]; i != -1; i = ne[i]) // 遍历以u为节点的子节点
	{
		int j = e[i]; // i对应存放的值
		if( !st[j]) // 检查是否遍历过
		{
			int s = dfs(j); // 得到子树的联通数量
			res = max(res, s); // 得到u节点子树中最大的节点数量
			sum += s; // 统计以u为根节点的子树共有多少个节点（包括本节点u） 所以这里sum初始值为1
			
		}
	}
	
	
	res = max(res, n - sum); // 这里统计的是删除节点u后的不同连通块的最大值
	
	ans = min(ans, res);// ans -- 遍历所有节点后的剩余各个连通块中点数的最大值最小值
	
	return sum;// 这里注意返回值问题 我们dfs传递的值应给是以u为几点的子树的总节点数 为了方便计算u节点的父亲节点对应的那部分连通块数量
		
}



int main()
{
	cin  >> n;
	memset(h, -1, sizeof h);
	for(int i = 0; i < n - 1 ; i ++)// 这里减去1是因为树的边比节点数少1
	{
		int a, b;
		cin  >> a >> b;
		// 这里为了方便回溯 引入无向边-- 即可以来回遍历父子节点
		add(a, b);
		add(b, a);
		
		
	}
	
	dfs(1);
	
	cout << ans << endl;
	return 0;
}