2017多校3 1005 RXD and dividing

http://acm.hdu.edu.cn/showproblem.php?pid=6060

求一棵树中从2-n的点分成k部分，求每一部分加上1这个根节点的最小生成树长度和的最大值。

可以理解成求每条边的贡献，最大边肯定是选择越多越好，那么我们就可以去寻找每一个根节点下的节点数与k的比较，如果k小，那该根节点的边就可以认为会贡献k次，如果是节点数小，那么就是过节点数次，dfs去寻找节点数就可以了。

#include<iostream>#include<cstring>#include<cstdio>using namespace std;const long long int  N=10000007;struct node{    int  en;    int v;    int next;}e[N];int  n,k;int  cnt;int  _hash[N];long long int  ans;void _push(int  op,int  ed,int  cost){     e[++cnt].en=ed;     e[cnt].v=cost;     e[cnt].next=_hash[op];     _hash[op]=cnt;    }int  find(int op,int ed,int cost){    int i;    int  sum=1;    for(i=_hash[ed];i;i=e[i].next)    {        int too=e[i].en;        if(too!=op)        sum+=find(ed,too,e[i].v);     }    if(sum>k)ans+=(long long int )cost*k;    else ans+=(long long int)cost*sum;    return sum;}int main(){    while(~scanf("%d%d",&n,&k))    {        ans=0;        cnt=0;        int  op,ed,cost;        int i;        memset(_hash,0,sizeof(_hash));        for(i=1;i<n;i++)        {            scanf("%d%d%d",&op,&ed,&cost);            _push(op,ed,cost);            _push(ed,op,cost);        }        find(0,1,0);        printf("%lld\n",ans);    }    return 0;}