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

题目链接:https://www.acwing.com/problem/content/847/

题目
给定一颗树，树中包含 $n$ 个结点（编号 $1\sim n$）和 $n-1$ 条无向边。

请你找到树的重心，并输出将重心删除后，剩余各个连通块中点数的最大值。

重心定义：重心是指树中的一个结点，如果将这个点删除后，剩余各个连通块中点数的最大值最小，那么这个节点被称为树的重心。

输入格式
第一行包含整数 $n$，表示树的结点数。

接下来 $n-1$ 行，每行包含两个整数 $a$ 和 $b$，表示点 $a$ 和点 $b$ 之间存在一条边。

输出格式
输出一个整数 $m$，表示将重心删除后，剩余各个连通块中点数的最大值。

数据范围
$1\le n\le 10^5$

输入样例：

9
1 2
1 7
1 4
2 8
2 5
4 3
3 9
4 6

输出样例：

4

思路:

如图所示，假设删除点 $4$ ，可以用DFS求从 $4$ 往下的连通块数量
$4$ 往上的可以用 $n-size$ 来求 (size就是从4往下的连通块内点的数量)

DFS的返回值就是当前点的size

AC代码

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

using namespace std;

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

int n,ans = N;
int h[N],e[M],ne[M],idx;
bool vis[N];

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


// 以u为根的子树中点的数量
int dfs(int u)
{
    vis[u] = true;
    int sum = 1,res = 0;
    for(int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i];
        if(!vis[j])
        {
            int s = dfs(j);
            res = max(res,s);
            sum += s;
        }
    }
    res = max(res,n - sum);
    ans = min(ans,res);
    return sum;
}


int main()
{
    cin >> n;
    memset(h,-1,sizeof(h));
    for(int i = 0; i < n - 1; i++)
    {
        int a,b;
        cin >> a >> b;
        add(a,b);
        add(b,a);
    }
    dfs(1);
    printf("%d\n",ans);
    return 0;
}