846. 树的重心

题目：846. 树的重心

代码：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<queue>
#include<cmath>

using namespace std;
const int N = 100010,M = N*2;
//有n个头节点 
int m;
int h[N],ne[M],e[M],idx;
bool st[N];

int ans = N;//存最小的最大值 

void add(int a,int b){//在头节点a的后面添加b 
	e[idx] = b;
	ne[idx] = h[a];
	h[a] = idx++;//先用后加 
}
//以u为根节点的子树的大小 
int dfs(int u){
	st[u] = true;
	int summ=1;//当前子树大小，自身节点也算一个 
	int res=0;//将该点删除后，剩下连通块中的最大值 
	for(int i=h[u];i != -1;i=ne[i]){//访问u的每一个子节点 
		int j = e[i];
		if(!st[j]){
			int s = dfs(j);//以j为根节点的子树的大小
			res = max(res,s);//当前的子树也算一个连通块，要和res取较大值 
			summ += s; //以u为根的包括以j为根的 
		}
	} 
	
	res = max(res,m-summ);
	ans = min(res,ans);
	return summ; 
}

int main()
{
	
	memset(h,-1,sizeof h);
	scanf("%d",&m);
	for(int i=0;i<m-1;i++){
		int a,b;
		scanf("%d%d",&a,&b);
		add(a,b);
		add(b,a);//无向图，加两条边 
	}
	dfs(1);
	cout<<ans<<endl;
	return 0; 
}