[AcWing]846. 树的重心（C++实现）树与图的dfs模板题

1. 题目

2. 读题（需要重点注意的东西）

思路：用深度优先搜索的思想，在每个节点处可以通过递归得到子节点总的数量，然后就可以求出除去每个节点后的最大连通子图的节点数量，最后输出最小的那个最大连通子图的节点数量即可。

如果对递归思想有疑问的同学，建议先划到下方，4. 可能有帮助的前置习题，在给出的链接中复习一下递归的思想。弄清楚递归是如何求得每个节点的子树的总的节点数量的。

我们将该问题分解为如下四个子问题：

1、树的重心的含义？
2、图是怎么存储的？
3、图是怎么遍历的？
4、怎么求除去该数后最大连通子图的节点数量？

---

1、树的重心的含义
有这样一幅无向图

去除1后，剩下的连通块数的最大值为6，有6个节点

同理，可以求出去除每个节点后，剩下的连通块数的最大值为，如下图红色标记

因此，剩下的连通块数的最大值最小为5，树的重心是2或者3

---

2、图是怎么存储的？
要想使用dfs来遍历树或图，首先要知道图是怎么存储的。
为了简便，我们在这里使用有向图来说明图的存储方式。
图一般用邻接表来存储，邻接表是由数组实现的
如有这样一个有向图

其相应的邻接表为

那么在加入一条边，在代码中如何实现呢？
其实其实质就是链表的插入，如在2节点后插入一条到8的边


其用代码实现就是

e[idx] = 8,ne[idx] = h[2],h[2] = idx ,idx ++ ;

抽象出来，在插入一条从a到b的边的代码就是

e[idx] = b,ne[idx] = h[a],h[a] = idx ,idx ++ ;

---

3、图是怎么遍历的？
现在我们知道如何插入一条边了，那么如何遍历这个邻接表呢？
对于上述的数组h，我们来尝试通过输入的节点u来遍历它

对于输入的节点u，起始节点是 h[u]，下一个节点是ne[u]，终止条件是 ne[u] = -1；
因此遍历的代码是

for(int i = h[u];i != -1; i = ne[i])

---

4、怎么求除去该数后最大连通子图的节点数量？

• 用sum保存输入的u节点的子树的节点数

int s = dfs(j); // s为j的子树的节点数
size = max(size, s); // ???
sum += s;

• 用size保存最大连通块的节点数量
  即子树的节点数和总节点减去子树的节点相比哪个大。
  （因为除去某个节点时，子树为一个连通块，总节点减去子树为另一个连通块，size要保存这两个连通块中较大的一个）

 size = max(size, s); // 子树也是一个连通块，也需要考虑在size中
 ...
 ...
 size = max(size, n - sum - 1);

• 用ans保存剩余各个节点的连通块数的最大值的最小值，只需要在每个节点求出最大连通块的数量时比较一下，取较小的那一个即可

ans = min(ans, size);

3. 解法

---------------------------------------------------解法---------------------------------------------------

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010, M = N * 2;

int n;
// h[N] 表示有N个槽
// e[M], ne[M], idx含义与单链表一致
int h[N], e[M], ne[M], idx;
int ans = N;
bool st[N]; // 表示哪些点已经遍历过了

// 插入一条a到b的边
void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}

// 深度优先搜索dfs
int dfs(int u)
{
    st[u] = true; // 当前节点已经被搜索过了

    int size = 0, sum = 0;
    for (int i = h[u]; i != -1; i = ne[i]) // 遍历数的所有边
    {
        int j = e[i];
        if (!st[j]){
            int s = dfs(j);
            size = max(size, s);
            sum += s;
            
        }
    }

    size = max(size, n - sum - 1);
    ans = min(ans, size);

    return sum + 1;
}

int main()
{
    scanf("%d", &n);

    memset(h, -1, sizeof h);

    for (int i = 0; i < n - 1; i ++ )
    {
        int a, b;
        scanf("%d%d", &a, &b);
        add(a, b), add(b, a);
    }

    dfs(1);

    printf("%d\n", ans);

    return 0;
}

4. 可能有帮助的前置习题

• [AcWing]842. 排列数字（C++实现）dfs模板题，递归思想的解释 （主要是其中的递归思想！！）

5. 所用到的数据结构与算法思想

• 递归
• dfs

6. 总结

dfs遍历树和图的模板题，需要好好理解递归的思想，推荐熟记全部代码。