AcWing 846. 树的重心

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10,M=2*N;
int n;
int h[N],e[M],ne[M],idx;
bool st[N];
//去掉重心的所有的子树中最大的子树的结点数目
int ans=N;
void add(int a,int b){
    e[idx]=b;
    ne[idx]=h[a];
    h[a]=idx++;
}
//sum：以u为根的子树中，点的数量
int dfs(int u){
    st[u]=true;
    int sum=1,res=0;
    for(int i=h[u];i!=-1;i=ne[i]){
        int j=e[i];
        if(!st[j]){
            //s：以j为根的当前子树中，点的数量，类比sum
            int s=dfs(j);
            //res:删掉j点后，当前所有连通块中点数的最大值
            res=max(res,s);
            //j是u的一部分
            sum+=s;
        }
    }
    //res:删掉u点后，当前所有连通块中点数的最大值
    res=max(res,n-sum);
    ans=min(res,ans);
    return sum;
}
int main(){
    cin>>n;
    memset(h,-1,sizeof h);
    for(int i=0;i<n-1;i++){//n为点的个数，只能连n-1条边
        int a,b;
        cin>>a>>b;
        add(a,b),add(b,a);
    }
    dfs(1);
    cout<<ans<<endl;
    return 0;
}