2017 Multi-University Training Contest - Team 3 1005 RXD and dividing

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6060

题目大意: 给n个点和n-1条边， 每条边有个权值， 现在让你把2~n个点分到k个集合中， 每个点只能分到一个集合中，  在每个集合中的点和一号点形成一个最小生成树， 现在让你分这n个点， 使得k个几个的最小生成树的权值和最大。

解题思路：  对于一条边， 我们求出这条边下面的点数sz， 如果sz， 如果sz > k  这条边就会给答案贡献k次， 反之 贡献sz次。 

//2017-08-02 13:02
//2017-08-02 13:29
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long LL;
typedef pair<int , int> PII;
const int MaxN = 1e6;

vector<PII> edge[MaxN + 5];
int sz[MaxN + 5], falen[MaxN + 5];
int n, k;
LL ans;

void dfs(int u, int fa, int l){
	falen[u] = l;
	sz[u]++;
	for(int i = 0; i < edge[u].size(); i++){
		int v = edge[u][i].first;
		int nxtl = edge[u][i].second;
		if(v != fa){
			dfs(v, u, nxtl);
			sz[u] += sz[v];
		}
	}
	ans = ans + (LL)falen[u] * (LL)min(sz[u], k);
}

int main(){
	while(~scanf("%d %d", &n, &k)){
		ans = 0;
		memset(sz, 0, sizeof(sz));
		memset(falen, 0, sizeof(falen));
		memset(edge, 0, sizeof(edge));
		for(int i = 1; i < n; i++){
			int u, v, c;
			scanf("%d %d %d", &u, &v, &c);
			edge[u].push_back(make_pair(v, c));
			edge[v].push_back(make_pair(u, c));
		}
		dfs(1, 0, 0);
		printf("%lld\n", ans);
	}
	return 0;
}

比赛的时候这道题没读懂，  也没怎么想， 结束之后听他们一讲题意才明白。