当前位置: 代码迷 >> 综合 >> Codeforces A.Linova and Kingdom (树的深度 / dfs)
  详细解决方案

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

热度:5   发布时间:2023-12-22 13:20:35.0

传送门

题意: 给你一棵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;
}