树的重心

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 const int N=1e5+10,M=2*N;
 4 int h[N],ne[M],e[M],idx;
 5 int n;
 6 bool st[N];
 7 int sz[N];
 8 int ans=N;
 9 int pos;
10 void add(int a,int b){
11    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
12 }
13 void dfs(int x){
14    st[x]=1,sz[x]=1;
15    int max_part=0;
16    for(int i=h[x];i!=-1;i=ne[i]){
17       int y=e[i];
18       if(st[y]) continue;
19       dfs(y);
20       sz[x]+=sz[y];
21       max_part=max(max_part,sz[y]);
22    }
23    max_part=max(max_part,n-sz[x]);
24    if(max_part<ans){
25       ans=max_part;
26       pos=x;
27    }
28 }
29 int main(){
30    memset(h,-1,sizeof h);
31    cin>>n;
32    for(int i=0;i<n;i++){
33       int a,b;
34       cin>>a>>b;
35       add(a,b),add(b,a);
36    }
37 
38    dfs(1);
39    cout<<ans<<endl;
40 }