【HDU6060】RXD and dividing(dfs)

记录一个菜逼的成长。。

2017 Multi-University Training Contest - Team 3

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

考虑每条边的贡献。
对于一条边$\left(fa[x],x \right)$
要使得这条边的贡献最大，要使以$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 second
typedef 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;
}