树的重心题解

树的重心

给定一颗树，树中包含 n 个结点（编号 1∼n）和 n−1 条无向边。

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

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

输入格式

第一行包含整数 n，表示树的结点数。

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

输出格式

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

数据范围

1≤n≤1e5 

输入样例

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

输出样例 

4

 代码如下（采用邻接表建树）

#include<bits/stdc++.h>
using namespace std;
#define rep(i, a, b) for (int i = a; i <= b; i++)
const int N = 1e5 + 10;
vector<int> g[N];
bool vis[N];
int n,ans=N;//n为总节点个数,ans为遍历各个假设重心后的连通块点数最小值
int dfs(int u)
{
    int sum = 1, res = 0;//sum以u为根节点的树的节点个数，res存储各个连通块点数最大值
    vis[u] = 1;
    rep(i,0,g[u].size()-1)
    {
        if(!vis[g[u][i]])
        {
            int s = dfs(g[u][i]);
            res = max(s, res);//回溯后更新res值
            sum += s;//回溯后以u为根结点的节点数
        }
    }
    res = max(res, n - sum);//跟另外一部分比较
    ans = min(ans, res);//更新最小值
    return sum;
}
int main()
{
    memset(vis, 0, sizeof(vis));
    int a, b;
    cin >> n;
    rep(i,1,n-1)
    {
        cin >> a >> b;
        g[a].push_back(b);
        g[b].push_back(a);
    }
    dfs(1);//从任一点开始遍历都可以
    cout << ans;
    return 0;
}