Codeforces A.Linova and Kingdom (树的深度 / dfs)

传送门

题意： 给你一棵n个节点的树，试标记其中k个节点为工业区，剩下的为旅游区，使得所有工业区走到1号根节点所经过的旅游区数量max。


思路：

• 将深度最大的k个节点标记为工业区即可。
• s[u]表示u节点子树上节点的数目。
• d[u]表示u节点的深度。
• 排序找到d[u]-s[u]最大的k歌节点即可。

代码实现：

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#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;
const int  inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 2e5 + 5;

int n, k, ans;
vector<int> g[N];
int s[N], d[N];

void dfs(int u, int f){
    s[u] = 1;
    for(auto v : g[u]){
        if(v == f) continue;
        d[v] = d[u]+1;
        dfs(v, u);
        s[u] += s[v];
    }
}

signed main(){
    IOS;

    cin >> n >> k;
    for(int i = 1; i < n; i ++){
        int u, v; cin >> u >> v;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    dfs(1, -1);
    for(int i = 1; i <= n; i ++) s[i] = d[i]-(s[i]-1);
    sort(s+1, s+n+1);
    for(int i = n; i > n-k; i --) ans += s[i];
    cout << ans << endl;

    return 0;
}