【AcWing】846. 树的重心(树与图的深度优先遍历)

无向图也是一种特殊的有向图。

树也是一种特殊的图。

遍历一下所有出边。

枚举删除每个点统计最大连通块的点数，点数最小的那个点就是要删除的重心。

树的深度优先遍历当中，能求出每个子树的点数。

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int n;
const int N=1e5+10,M=N*2;//N个点，开2N个边
int h[N],e[M],ne[M],idx;//h是多个链表。每个单链表是从左往右存储的，链表的右边是头，左边是尾。
bool st[N];

int ans=N;

void add(int a,int b){//建立单向边，从a至b
    e[idx]=b;
    ne[idx]=h[a];
    h[a]=idx++;
}
//以u为根的子树点的个数
int dfs(int u){
    st[u]=true;//标记搜过
    
    int sum=1,res=0;
    for(int i=h[u];i!=-1;i=ne[i]){//遍历u的所有子节点
        int j=e[i];
        if(!st[j]){
            int s=dfs(j);
            res=max(res,s);//比较当前值和以前记录的子节点的最大值
            sum+=s;//计算u节点所统领的所有子节点
        }
    }
    res=max(res,n-sum);//n-sum表示的是整个树删除以这个点为根的子树(表示删掉这个点的情况下其余部分的点数)
    ans=min(ans,res);//(题目)如果将这个点删除后，剩余各个连通块中点数的最大值最小
    return sum;//返回u节点+所有u节点统领的节点的点数和
}

int main(){
    cin>>n;
    memset(h,-1,sizeof(h));//初始化头结点，第一次的-1赋给ne[idx]做为尾
    for(int i=0;i<n-1;i++){
        int a,b;
        cin>>a>>b;
        add(a,b);add(b,a);
    }
    dfs(1);
    cout<<ans<<endl;
    return 0;
}