【ACwing】三、 搜索与图论： 树与图的深度优先遍历——846. 树的重心

（1）知识点

（2）树与图的深度优先遍历模板

建立邻接表：

//邻接表
int h[N], e[N * 2], ne[N * 2], idx;

void add(int a, int b) {
    e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

遍历模板：

// 需要标记数组st[N],  遍历节点的每个相邻的便
void dfs(int u) {
    st[u] = true; // 标记一下，记录为已经被搜索过了，下面进行搜索过程
    for (int i = h[u]; i != -1; i = ne[i]) {
        int j = e[i];
        if (!st[j]) {
            dfs(j);
        }
    }
}

（2）题目：

原题链接：https://www.acwing.com/problem/content/848/

（3）题目理解

如：

（4）思路：

用ans来记录最终结果（最小的各结点最大连通数），res记录删除节点u后最大的连通数。
将图的各边进行存储后，对节点u进行dfs，直到所有的点都被遍历过，此时就能得到重心ans。dfs函数返回以值u为根的子树大小，首先我们要知道当前节点的子树的大小sum（sum=两子树大小+1），然后取max(两子树大小,n-sum)的值赋给res（最大连通数），更新ans为min(ans,res);

（5）代码

# include <iostream>
# include <iostream> //因为memset
#include <cstring>
using namespace std;

const int N = 1e5 + 10;//数据的范围
const int M = 2 * N;

int n;
int h[N]; //邻接表存储树，有n个节点，所以需要n个队列头节点
int e[M];//按下标存储节点的值
int ne[M];//按下标存储各节点next的下标
int idx;//链表指针
void add(int a,int b)//添加一条a到b的边
{
    e[idx]=b;//将新下标的节点值赋值为b
    ne[idx]=h[a];//通过next将新节点与以a为头的链表产生关联
    h[a]=idx++;//头节点的下标更新为idx，同时idx++
}
int ans = N;//记录最终结果
bool st[N];//标记当前节点有没有走过 
int dfs(int u)//返回以值u为根的子树大小
{
    int sum = 1;//记录以u为根的子树大小，现在只有根所以为1
    int res = 0;//记录删除u后连通块的最大值
    st[u]=true;//当前节点被走过了
    for(int i=h[u];i!=-1;i=ne[i])//i表示值为u的链表表头下标，只要能找到next值就执行循环
    {
        int j = e[i];//j为下标为i的节点值
        if(!st[j])//若该节点值没被访问过
        {
            int  s = dfs(j);//进入下一层，返回单棵子树的大小
            res = max(res,s);//res为删除u后两子树中最大的
            sum+=s;//更新以u为根的子树大小
        }
    }
    res = max(res,n-sum);//求出父节点那坨和子节两坨中的最大值
    ans=min(res,ans);//更新最终结果为所有res中最小的那个值
    return sum;//返回以u为根的子树大小
}
int a,b;
int main(){
    memset(h,-1,sizeof(h));//初始化h数组
    cin>>n;
    //n个节点，n-1条边即n-1行数据
    for(int i=0;i<n-1;i++){
        cin>>a>>b;
        add(a,b),add(b,a);
    }
    dfs(1);//从值为1的节点开始dfs
    cout<<ans<<endl;
    return 0;
}

复盘出错位置

一定要记得在main函数里面对h数组进行初始化为-1！！！