【HDU6060】RXD and dividing(dfs)

记录一个菜逼的成长。。

2017 Multi-University Training Contest - Team 3

题目链接
题目大意:
给你一棵n个点的树，将2…n分成k个集合，每个集合不相交。
求每个集合Si⋃{1}后的f(S)值
f(S)表示添加一些边使得S集合中的点连通，并且使得边的权值和最小

考虑每条边的贡献。
对于一条边(fa[x],x)
要使得这条边的贡献最大，要使以x为父节点的节点(包括x节点本身)尽量分给不同集合。这个值是min(k,son[u]).

#include <stdio.h>#include <math.h>#include <string.h>#include <string>#include <vector>#include <algorithm>using namespace std;#define rep(i,l,r) for( int i = l; i <= r; i++ )#define rep0(i,l,r) for( int i = l; i < r; i++ )#define ALL(v) (v).begin(),(v).end()#define cl(a,b) memset(a,b,sizeof(a))#define clr clear()#define pb push_back#define mp make_pair#define fi first#define se secondtypedef long long LL;typedef pair<int,int> PII;const int INF = 0x3f3f3f3f;const int maxn = 1000000 + 10;vector<PII>edge[maxn];int son[maxn];LL ans;int n,k;void dfs(int u,int fa){  son[u] = 1;  for( int i = 0; i < edge[u].size(); i++ ){    int v = edge[u][i].fi,w = edge[u][i].se;    if(v == fa)continue;    dfs(v,u);    int sum = 0;    sum += son[v];    ans += (LL)w * min(k,sum);    son[u] += sum;  }}int main(){  while(~scanf("%d%d",&n,&k)){    cl(son,0);    for( int i = 1; i <= n; i++ )edge[i].resize(0);    for( int i = 1; i < n; i++ ){      int u,v,w;      scanf("%d%d%d",&u,&v,&w);      edge[u].pb(mp(v,w));      edge[v].pb(mp(u,w));    }    ans = 0;    dfs(1,0);    printf("%lld\n",ans);  }  return 0;}