CodeForces - 1337C Linova and Kingdom

题目链接：https://vjudge.net/problem/CodeForces-1337C
题目大意：有一颗n个节点的树，每个节点代表一个城市，让你选择k个城市，使这几个城市到根节点的路上 经过的不是这k个节点的节点数最多 求这k个点到根结点路上经过的总结点数

就比如样例3，如果5属于这k个数，那他堵了4的路 他的贡献就是他的深度-1

如果7属于这k个数，那他堵了4 8 5 3 的路（因为贪心7属于k那么8 4 3 5 必然属于k ) 他的贡献就是他的深度-4

我们发现，减掉的数就是以他为根的子树的大小

所以用一个结构体存下 深度和子树的大小 贡献=深度-子树大小

再按照贡献的大小排序，取前k大

sum应该是要开long long

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=2e5+10;
vector<int>v[maxn];
int vis[maxn];
struct tree
{
    int d;   ///深度
    int s;    ///节点的子树个数
    int mon;  ///d-s(每个节点的贡献)
}tr[maxn];
void dfs(int u,int pre)    ///dfs
{
    tr[u].d=tr[pre].d+1;
    tr[u].s=1;
    vis[u]=1;
    for(int i=0;i<v[u].size();i++)
    {
        int y=v[u][i];
        if(vis[y]==0)
        {
            dfs(y,u);
            tr[u].s+=tr[y].s;
        }
    }

}
int cmp(tree a,tree b) 
{
    return a.mon>b.mon;
}
int main()
{
    memset(vis,0,sizeof(vis));
   int n,k;
   scanf("%d%d",&n,&k);
   for(int i=1;i<=n-1;i++)  ///输入
   {
       int x,y;
       scanf("%d%d",&x,&y);

       v[x].push_back(y);
       v[y].push_back(x);

   }
   dfs(1,0);
   for(int i=1;i<=n;i++)
   {
       tr[i].mon=tr[i].d-tr[i].s;///每个节点的贡献
   }
   ll sum=0;
   sort(tr+1,tr+1+n,cmp); ///排序
   for(int i=1;i<=k;i++) ///取前k大的数的和
   {
       sum+=tr[i].mon;
   }
   printf("%lld\n",sum);
}