dfs : 树的重心

题目入口
给定一颗树，树中包含n个结点（编号1~n）和n-1条无向边。
请你找到树的重心，并输出将重心删除后，剩余各个连通块中点数的最大值。
重心定义： 重心是指树中的一个结点，如果将这个点删除后，剩余各个连通块中点数的最大值最小，那么这个节点被称为树的重心。
输入格式
第一行包含整数n，表示树的结点数。
接下来n-1行，每行包含两个整数a和b，表示点a和点b之间存在一条边。
输出格式
输出一个整数m，表示重心的所有的子树中最大的子树的结点数目。
数据范围
1 ≤ n ≤ 10^5
输入样例
9
1 2
1 7
1 4
2 8
2 5
4 3
3 9
4 6
输出样例：
4
思路：显然这个题需要求子树的节点数，就需要用到树的深度搜索了。用dfs(x)求到以x为根节点的树的节点数。
对于以u为重心的情况下，s[i], s[i + 1],便是其所有子树的节点数，而其父亲节点所在的树的节点数变为n - 1 - sum(即所有子树s,的和)，而答案ans = min(ans, max(n - 1 - sum, s中的max).
例如此以2位根节点的示例图：

代码实现：

#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <cctype>
#include <cstring>
#include <cmath>
#include <iostream>
#include <sstream>
#include <string>
#include <list>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <algorithm>
#include <functional>
#define int long long
#define lowbit(x) (x &(-x))
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pll;
const int  INF = 0x3f3f3f3f;
const double PI = acos(-1.0);
const double EXP = 1e-8;
const ll   MOD = 1e9 + 7;
const int  N = 1e5 + 5;
int n, ans = N;
int h[N], e[N << 1], ne[N << 1], tot;
bool vis[N];
//领接表的存储
void add(int a, int b)
{
    e[tot] = b;
    ne[tot] = h[a];
    h[a] = tot ++;
}
//计算以u为根节点子树的节点数
int dfs(int u)
{
    vis[u] = 1;
    int sum = 1, res = 0;  //sum表示当前节点数,res表示连通块点数的max
    for(int i = h[u]; ~i; 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;  //记住返回的一定是子树的节点数sum
}

signed main()
{
   IOS;
   cin >> n;
   int a, b;
   mem(h, -1);  // 必须要先将每个节点的头结点指向-1
   for(int i = 1; i < n; i ++){
      cin >> a >> b;
      add(a, b);
      add(b, a);  //无向边需要存两个方向
   }
   dfs(1);  //当然是从更节点开始遍历
   cout << ans << endl;
   return 0;
}