Codeforces Linova and Kingdom(树形结构、贪心）

题目大意：
给定一颗有n个节点的树，从中选出k个节点，从每个节点走到根节点1，路径中经过的未选中的节点的个数为kx，求一种选择方案，使总的kx之和最大。

解题思路：
求出每个节点对总值的贡献值，加入选择节点x，那么该节点的贡献值为x的深度减去该节点所有子树中节点的个数，因为选择节点的时候，必然使从树最深的地方开始的，当一个可以选择一个节点时，其子节点也必然已经被选择（贪心），选入该节点后，其子树中所有节点的贡献值减1
所有根据dfs求出所有节点的贡献值，然后排序，求和

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=2e5+10;
ll ans;
int n,k,sz[maxn];
vector<int> e[maxn],arr;
void dfs(int u,int f,int d)
{
    for(auto v:e[u])
        if(v!=f) dfs(v,u,d+1),sz[u]+=sz[v];
    arr.push_back(d-sz[u]);
    sz[u]++;
}
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=0;i<n-1;++i)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        e[x].push_back(y);
        e[y].push_back(x);
    }
    dfs(1,0,0);
    sort(arr.begin(),arr.end());
    for(int i=n-1;k;i--,k--)
        ans+=arr[i];
    printf("%lld",ans);
    return 0;
}