题目链接: 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;
}