树的重心（dfs枚举）

树的重心（dfs枚举）

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

const int N = 1e5 + 10, M = N*2;
int h[N];
struct node{
    int to,nex;
}e[M];
bool st[N] = {0};   //visted数组记录已经访问的元素; c++有一个zero-initializer(零初始化器)
int idx = 1;
int n;
int ans = N; //ans是要循环找到的最小值，所以初始化为最大值

void add(int u, int v){
    e[++idx] = (node){v, h[u]};
    h[u]=idx;
}


int dfs(int u){
    st[u] = true;
    int sum = 1;  //sum是记录这个u为根的树的节点总数，当前有一个初始化为1
    int res = 0; //res是删除u后所有连通块的最大值，所以初始化为最小值，这个res是对这个u而言的
    
    for(int i=h[u]; i!=-1; i=e[i].nex){
        int j = e[i].to;
        
        if(!st[j]){
            int s = dfs(j);   //此处假设算法成功，我们已经求出了其中一个子树的节点个数，也是其中一个连通块的大小
            res = max(res, s); //不断更新找所有连通块里的最大值，后面的for层都会不断更新，找到“子树连通块”中的最大值
            sum += s;   //j是u的一个子树，加到u的总数里面，后面的for层都会依次加入，求的u的sum
        }
    }
    
    //for循环结束
    //目前我们已经求出u为根的树的节点总数sum；删除u后“子树连通块”中的最大值res. 
    res = max(res, n - sum);  //除了“子树连通块”还要与“父亲连通块”的大小来比较，“父亲连通块”通过减法求出 (对于结束节点，最大值就是n-1)
    ans = min(ans, res);  //找所有不同u的最小值，每个节点都会走一次dfs
    
    return sum;   //函数返回u为根的树的节点总数
}


int main(){
    memset(h, -1, sizeof h);
    cin>>n;
    for(int i=0;i<n;i++){
        int u, v;
        cin>>u>>v;
        add(u,v);
        add(v,u);    //因为树是无向边
    }
    
    dfs(1);
    cout<<ans<<endl;
    return 0;
}