洛谷 P5536 【XR-3】核心城市

题目链接：

【XR-3】核心城市 - 洛谷

思路：

两种做法：

1.拓扑排序，这种做法个人认为更好理解，思路解析参见P5536 - wheneveright 的博客 - 洛谷博客

2.求树的直径，然后贪心地选择【离核心城市群最远的城市】，往那个方向放一个核心城市（这样距离就-1了）。实现方法是计算每个点的【子树深度】，优先选择在子树深度大的地方放核心城市。思路解析参见：P5536 【XR-3】核心城市（直径）_Robin2321的博客-CSDN博客

代码：

1.拓扑排序：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
int n, k, a, b;
vector<int> g[maxn];
int in[maxn];
queue<int> q;

int main(){
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    cin >> n >> k; int cnt = n;
    for(int i=1; i<=n-1; i++){
        cin >> a >> b;
        g[a].push_back(b);
        g[b].push_back(a);
        in[a]++; in[b]++;
    }
    for(int i=1; i<=n; i++){
        if(in[i] == 1) q.push(i);
    }
    int ans = 0;
    while(q.size()){
        int size = q.size();
        cnt -= size; ans++;
        if(cnt <= k) {cout<<ans; return 0;}
        while(size--){
            int u = q.front(); q.pop();
            for(int v : g[u]){
                if(--in[v] == 1) q.push(v);
            }
        }
    }
}

2.树的直径：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
int n, k, a, b;
vector<int> g[maxn];
int d[maxn], dis[maxn]; //深度,子树深度
int pre[maxn]; //记录前驱结点（路径记录）
int farL=0, farI=-1; //最远距离和下标

bool cmp(int x,int y) {return x>y;}
void dfs(int u, int fa, int get){ //起点，父节点，是否记录路径
    if(d[u]>farL) {farL=d[u]; farI=u;}
    for(int v : g[u]){
        if(v==fa) continue;
        if(get) pre[v] = u;
        d[v] = d[u]+1;
        dfs(v, u, get);
    }
}
void getDis(int u, int fa){
    dis[u] = 1;
    for(int v : g[u]){
        if(v==fa) continue;
        getDis(v, u);
        dis[u] = max(dis[u], dis[v]+1);
    }
}
int main(){
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    cin >> n >> k;
    for(int i=1; i<=n-1; i++){
        cin >> a >> b;
        g[a].push_back(b);
        g[b].push_back(a);
    }
    dfs(1,0,0); //从结点1出发，找最远点（0表示无效父节点）
    d[farI] = farL = 0; //最远点的d记为0，找直径
    dfs(farI,0,1); //找直径（farI的最远点）
    int mid = farI; //中点，从farI开始倒序寻找
    for(int i=1; i<=(farL+1)>>1; i++) mid = pre[mid]; //寻找中点位置
    getDis(mid, 0);
    sort(dis+1, dis+n+1, cmp);
    cout << dis[k+1];
}