AcWing846 树的重心(DFS、树和图的遍历)

题目：AcWing846 树的重心

---

前言

树，即连通无环图，它是一种特殊的图。无向图是特殊的有向图。邻接矩阵和邻接表都可以存储图。邻接矩阵适用于稠密图。领接表适用于稀疏图。我们可以用DFS和BFS遍历图的存储结构。

一、题目陈述

给定一颗树，树中包含n个结点（编号1~n）和n-1条无向边。
请你找到树的重心，并输出将重心删除后，剩余各个连通块中点数的最大值。
重心定义：重心是指树中的一个结点，如果将这个点删除后，剩余各个连通块中点数的最大值最小，那么这个节点被称为树的重心。
输入格式
第一行包含整数n，表示树的结点数。
接下来n-1行，每行包含两个整数a和b，表示点a和点b之间存在一条边。
输出格式
输出一个整数m，表示将重心删除后，剩余各个连通块中点数的最大值。
数据范围
1≤n≤1e5
输入样例
9
1 2
1 7
1 4
2 8
2 5
4 3
3 9
4 6
输出样例
4

二、解决思路

我们可以用DFS从某一个节点（比如节点1）开始遍历整棵树，DFS遍历的实用处之一在于可以在遍历的过程中统计子树包含的节点的个数。我们可以利用这一特性，计算将某一节点作为重心去除后，各个子树以及父结构包含的节点的个数的最大值。对每一个点进行统计后，即可确定重心以及最小的最大包含点数。

三、代码实现

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 1e5+10,M=N*2;
// n个点 m条边 
int n,m;
// 全局答案，即最小的最大值，初始值超级大 
int ans = N; 
// 领接表存图  idx和节点编号从0还是1开始没有任何关系，idx没有实用作用 
int h[N],e[M],ne[M],idx;
// 头插法 将节点b插入到节点a伸出的链表中 
void add(int a,int b) {
	e[idx] = b;
	ne[idx] = h[a];
	h[a] = idx++; 
} 
// 遍历图用的状态数组 
bool st[N];
// 遍历以u为根节点的树  返回这棵树的节点个数包括根节点 
// 这个写法中，并没有恢复现场，所以所有的点都只会被访问到1遍，那么从节点1开始向下搜索时，
// 必然要用n-num得出父结构包含点的个数 
int dfs(int u) {
	// num记录以u为根节点的树包含的节点总数，包括节点u 
	int num = 0; 
	// sonmax记录最大子树包含的节点个数 
	int sonmax = -2e9;
	// 标记一下已经被搜过了 
	st[u]=true;
	num ++; 
	for(int i=h[u];i!=-1;i=ne[i]) {
		int j=e[i];
		if(!st[j]) {
			int temp = dfs(j);
			num += temp;
			sonmax = max(sonmax, temp);
		}
	}
	// leftmax记录去掉这个点后，最大部分包含的点数 
	int leftmax = max(sonmax, n-num);
	ans = min(ans, leftmax);
	return num;
}
int main() {
	memset(h,-1,sizeof(h));
	cin>>n;
	m = n - 1;
	for(int i=0;i<m;i++) {
		int a,b;
		cin>>a>>b;
		add(a,b);	add(b,a);
	}
	dfs(1);
	cout<<ans<<endl;
	return 0;
}

总结

要善于利用DFS在搜索过程中可以统计树包含节点个数的特性进行编码。无向图在建立数据结构存储的时候，可以添加两条方向相反的边。稀疏图常用领接表存储，所谓稀疏图是指m远小于n方。