树的重心dfs

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

#include <bits/stdc++.h>
using namespace std; 
const int N=1e6+10;
int e[N],ne[N],h[N],idx;
int n,pos;//pos为节点编号 
int ans=N;//表示重心的所有子树中最大的子树的节点数目 
int size[N];
int v[N];//标记 
void add(int a,int b){
	e[idx]=b;
	ne[idx]=h[a];
	h[a]=idx++;
}
void dfs(int x){
	v[x]=1;
	size[x]=1;
	int mx=0;
	for(int i=h[x];i!=-1;i=ne[i]){
		int y=e[i];
		if(v[y])continue;
		dfs(y);
		size[x]+=size[y];
		mx=max(mx,size[y]);
	}
	mx=max(mx,n-size[x]);
	if(mx<ans){
		ans=mx;
		pos=x;
	}
}
int main(){ 
	cin>>n;
	memset(h,-1,sizeof(h));
	for(int i=1;i<n;i++){
		int a,b;
		cin>>a>>b;
		add(a,b);
		add(b,a);
	}
	dfs(1);
	cout<<ans<<endl;
	return 0;
}