树的重心(dfs)

#include <bits/stdc++.h>
using namespace std;
const int N = 1000, M = N * 2;
int ans = N; //最小最大值
int n;
int h[N], e[M], ne[M], idx; //使用邻接表来存树
bool st[N];//标记数组

void add(int a, int b) {
	//表示将b插入到a的后面
	e[idx] = b;
	ne[idx] = h[a];
	h[a] = idx++;
}

//返回以u为根的子节点的数量
int dfs(int u) {
	st[u] = true;
	int sum = 1, res = 0; //sum表示u结点的子树大小，res表示把u这个点删除之后产生的连通块中的最大值
	for (int i = h[u]; i != -1; i = ne[i]) {
		int j = e[i];
		if (!st[j]) {
			int s = dfs(j); //u的儿子j的大小
			res = max(res, s); //统计最大的连通块
			sum += s;
		}
	}
	res = max(res, n - sum);//比较完儿子们的连通块之后比较父辈的连通块取最大值
	ans = min(ans, res); //取最小最大连通块
	return sum;
}

int main() {
	memset(h, -1, sizeof h);
	cin >> n;
	for (int i = 0; i < n - 1; i++) {
		int a, b;
		cin >> a >> b;
		add(a, b);
		add(b, a);
	}
	dfs(1);
	cout << ans;
	return 0;
}