acwing树的重心

给定一颗树，树中包含 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

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

using namespace std;
const int N = 100010 , M = N * 2;
int n;
int h[N];// n个链表的链表头
int e[M];//存的是每一个节点的值为多少
int ne[M];//存的是每一个节点的next指针是多少
int idx;
//深度优先遍历和宽度优先遍历  每个节点都只会遍历一次
bool st[N];//去记录那个点已经遍历过了
int ans = N;



//向a头节点的后面插入b
void add(int a, int b)
{
	e[idx] = b;
	ne[idx] = h[a];
	h[a] = idx;
	idx++;
}

//以u为根的子树中点的数量
int dfs(int u)
{
	st[u] = true;//先标记一下当前这个点已经被搜索过了

	//先记录一下当前这个子树的大小,当前这个点算是一个点 所以是从1开始
	int sum = 1; 

	int res = 0;//记录的是 把该点删除之后 每一个连成块的最大值



	//遍历一下u的所有的初边
	//一个节点下面的子树
	for (int i = h[u]; i != -1; i = ne[i])
	{
		int j = e[i];//当前这个链表里面的节点  对应的图里面的编号是多少

		if (!st[j])
		{
			// 如果j没有被搜索到 dfs(j)  一条路走到黑
			int s = dfs(j);//用s来表示子树的大小

			res = max(res , s);

			sum += s;//s表示子树的大小  加上原本被删除的节点
		}
	}
	//去掉删除的节点以及子树剩下的节点
	res = max(res,n - sum);

	ans = min(ans, res);

	return sum;
}

int main()
{
	cin >> n ;
	//链表的初始化  让n个节点全部指向-1
	memset(h,-1,sizeof h);

	for (int i = 0; i < n -1; i++)
	{
		int a, b;
		cin >> a >> b;
		add(a, b), add(b, a);//因为是无向边
	}

	dfs(1);//随便挑选一个点  比如从第一个点开始搜索

	cout << ans << endl;
	return 0;
}