hdu6060 RXD and dividing

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6060
题意：给你一棵有n个结点的树，让你把他的2~k个节点分成k份，然后求1号顶点到每份里各个顶点的最大权值和，比如说样例，就是分成了{2},{3},{4},{5}，到每个顶点的距离即是到这个顶点在原图上到顶点的距离
解析：其实画多几棵树会发现一个问题，就是有些变权是会被多次重复计算的，比如样例，(1,2)这条边就重复计算了3次，所以肯定最好的结果，是把2的子节点尽可能不和2分到一起，那么(1,2)这条边最多计算几次呢，当然是min(k,sz[2])，sz[2]表示2的子树规模大小，按这个规律去从根节点dfs一次，预处理处每个结点的子树大小，答案应该就是每个边权乘上其对应的权值求和

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6+100;
typedef long long ll;
vector<pair<int,int> >G[maxn];
int sz[maxn];
int w[maxn];
void dfs(int u,int fa)
{
    sz[u] = 1;
    for(int i=0;i<(int)G[u].size();i++)
    {
        pair<int,int> v = G[u][i];
        if(v.first==fa)
            continue;
        w[v.first] = v.second;
        dfs(v.first,u);
        sz[u] += sz[v.first];
    }
}
int main(void)
{
    int n,k;
    while(~scanf("%d %d",&n,&k))
    {
        for(int i=0;i<=n;i++)
            G[i].clear();
        memset(sz,0,sizeof(sz));
        memset(w,0,sizeof(w));
        for(int i=0;i<n-1;i++)
        {
            int u,v,w;
            scanf("%d %d %d",&u,&v,&w);
            G[u].push_back(make_pair(v,w));
            G[v].push_back(make_pair(u,w));
        }
        dfs(1,-1);
        ll sum = 0;
        for(int i=2;i<=n;i++)
            sum += 1LL*w[i]*min(k,sz[i]);
        printf("%I64d\n",sum);
    }
    return 0;
}