2017 - 多校训练-3 - 1005 RXD and dividing

题目链接: hdu 6060 RXD and dividing

题意:  给定n个点, n-1条边 （所给边为无向, 由这里其实已经看出是最小生成树了）, 1为根节点, 将点2-n划分k个块, 求每个块并根结点1所构成的生成树的权值之和.

给出的样例解释如下:

n=5,  k=4

划分成k块, 那其实就是2-5,每个节点单独成 块,则形成四个生成树如下

    

 其实我们可以从边u->v单独考虑,  如果当前点v的子树(以该点为根,往下形成一棵树)中包含x个块，那么边u->v对最终的答案贡献的价值为 x*w,  w为u->v的边权.  可以理解为x个块所构成的生成树都经过该边,故该边计算了x次.

尽可能保证x最大,那么其实x = min( cnt, k),  cnt表示当前v的子树的节点个数, k表示题目所给的划分块数.

code:

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#define mp make_pair
using namespace std;
typedef __int64 (ll);
int len=0;
int n, k;
const int maxn = 1e6+5;
int mark[maxn];
// vector<int>mp[maxn];
ll ans=0;
vector<  pair<int, int> > e[maxn];

void dfs(int u, int w){
	++len;
	mark[u] = 1;
	int p = len;
	int i, j;
	for(i=0;i<e[u].size();i++){
		int v = e[u][i].first;
		int w = e[u][i].second;
		if(mark[v]) continue;
		dfs(v, w);
	}
	ans+=(ll)w*min(k, len-p+1);
	return ;
}
int main(){
	int i, j;
	int u, v, w;
	while(~scanf("%d %d", &n, &k)){
		len=0;
		ans=0;
		memset(mark, 0, sizeof(mark));
		for(i=0; i<=n; i++) e[i].clear();
		for(i=0; i<n-1; i++){
			scanf("%d %d %d", &u, &v, &w);
			e[u].push_back(mp(v, w));
			e[v].push_back(mp(u, w));
		}
		dfs(1, 0);
		printf("%I64d\n", ans);
	}
	return 0;
}