Acwing 846 树的重心

思路

用vector存图 依次遍历删掉每个节点以后剩余连通块的最大值

code

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int N = 1e5 + 10;
bool st[N];
vector<int> v[N];
int n;
int ans = N;

// 以u为根的点的数量
int dfs(int u) {
    st[u] = 1; // 标记当前点被搜过了
    
    int sum = 1, res = 0; // sum表示当前子树的大小 
    for (int i = 0; i < v[u].size(); i ++ ) {
        int k = v[u][i];
        if (!st[k]) {
            int s = dfs(k); // 当前子树的大小
            res = max(res, s); // 当前子树也算连通块
            sum += s;
        }
    }
    res = max(res, n - sum); // 和上面的一块取最大值
    ans = min(ans, res);
    
    return sum;
}

int main() {
    cin >> n;
    for (int i = 0; i < n - 1; i ++ ) {
        int a, b;
        cin >> a >> b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    dfs(1);
    
    cout << ans << "\n";
    
    return 0;
}