Codeforces Round #635 (Div. 2)

这里单纯写下C题的题解，后期有时间写下别的题

传送门

C.Linova and Kingdom

涉及知识点：

• dfs、思维、贪心

解题思路：

• 根据题意我们了解到，我们尽量选距离根节点1较远的点作为工厂，肯定会考虑到每个点距根节点的距离，这里利用dfs来搜索进行存储，然后我们如何选择点成为了关键
• 这里我们想一下，如果当前点作为工厂，那么他的贡献度为 ： 当前点的深度 - 前面的黑色的点（就是工厂）
• 那么我们转换一下，我们的贪心思想是 当前点的深度 - 其子节点的数目 （deep - size），然后从大到小选取就OK
• 首先证明一下一个这个贪心，首先是证明深层的点，他的贡献度就是他的深度，这个很容易想到，如果一个点不是最底层的一个点，他的下面的点的贡献度都是 deep - size

• 第二层的点，如果选为工厂，那么他的贡献度 deep - size 就是 2 - 2,是0， 但是可以看出他的贡献率应该为2，这是因为，选择这个点，那么他下面的2个点也肯定是会选的，如果当前这个点选了，他的两个子节的贡献度都会减少 1 ，所以有多少子节点，如果选这个点当工厂，那么的贡献度就会减少多少（可以理解一下）
• 然后按照这个规律从大到小进行排序，我们选出的数值相加就是最大的贡献度

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long ll;

const int N = 200010, M = 400010;

int h[N], ne[M], e[M], ans[M], dist[M], res[N], idx;

void add(int x, int y){
    e[idx] = y, ne[idx] = h[x], h[x] = idx++;
}

void dfs(int u, int f){
    ans[u] = 1;
    for (int i = h[u]; i != -1; i = ne[i]){
        int j = e[i];
        if (j != f){
            dist[j] = dist[u] + 1;
            dfs(j,u);
            ans[u] += ans[j];
        }
    }
}

int main(){
    int n, k;
    scanf("%d%d",&n,&k);
    memset(h,-1,sizeof h);
    int x, y;
    for (int i = 1; i < n; i++){
        scanf("%d%d",&x,&y);
        add(x,y), add(y,x);
    }
    dist[1] = 1;
    dfs(1,0);
    for (int i = 1; i <= n; i++){
        res[i] = dist[i] - ans[i];
    }
    sort(res + 1,res + 1 + n);
    ll cnt = 0;
    for (int i = n ; i >= n - k + 1; i --){
        cnt += res[i];
    }
    printf("%lld\n",cnt);
    return 0;
}