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;
}