树的重心问题

树的重心：对于任意一个节点x,如果我们把它从树中删除，那么原来的一棵树可能分成若干个不相连的部分，其中每一部分都是一棵子树，设max_part（x)表示在删除x后的最大的子树大小，使max_part取到最小值的节点p就称之为树的重心
例题：acwing 846

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

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

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

输入格式第一行包含整数 n，表示树的结点数。
接下来 n−1行，每行包含两个整数 a和 b，表示点 a和点 b之间存在一条边。

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

数据范围
1≤n≤105
输入样例
9
1 2
1 7
1 4
2 8
2 5
4 3
3 9
4 6
输出样例：
4

#include <bits/stdc++.h>
using namespace std;
const int M=1e5+100;
const int N=2*M;
int n,head[N],nxt[N],ver[N],v[N],s[N],pos,ans,tot;
void add(int a,int b){
	ver[++tot]=b;
	nxt[tot]=head[a];
	head[a]=tot;
} 
void dfs(int x){
	v[x]=1;//访问标记 
	s[x]=1;//叶子节点的大小为1 
	int max_part=0;
	for(int i=head[x];i;i=nxt[i]){
		int y=ver[i];
		if(v[y])continue;
		dfs(y);
		s[x]+=s[y];	//根节点的子树大小为各子树之和 
		max_part=max(max_part,s[y]); //记录各子树的最大值 
	}
	max_part=max(max_part,n-s[x]);//x的父节点的另一边子树 
	if(max_part<ans){
		ans=max_part;//记录最大的子树大小 
		pos=x;//记录重心 
	} 
}
int main(){
	cin>>n;
	for(int i=1;i<n;i++){
		int a,b;
		cin>>a>>b;
		add(a,b);
		add(b,a);
	}
	ans=N;
	dfs(1);
//	cout<<ans;
	cout<<pos<<" "<<ans;
	return 0;
}