2017 Multi-University Training Contest 3 1005

  开始被题意打了，以为很难，什么斯坦纳树的，结果是一个很简单的dfs的题。

题意：让你找一个集合序列S让每一个集合加上1这个节点，然后使他们互相连通的最小生成树的权值和最大。

解法：就是尽量使每个节点的儿子分配为一个集合，这样保证尽量让父亲到根节点1的边反复用，为从1开始dfs，把每个节点的儿子数找到，然后乘以这个节点到上个节点的边权值。计算总和就是答案了。

#include <stdio.h>
#include <stdlib.h>
#include <cmath>
#include <string.h>
#include <string>
#include <algorithm>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <vector>
#include <iostream>
#define LL long long
#define INF 0x3f3f3f3f
const int MAX_N = 1e6+10;
const int mod = 100;
const double eps = 1e-6;
using namespace std;
int n,k;
struct node
{
    int to,next;
    LL w;
}es[MAX_N*4];
int e,head[MAX_N],son[MAX_N];
void addedge(int u,int v,LL w)
{
    es[e].to = v;
    es[e].w = w;
    es[e].next = head[u];
    head[u] = e++;
}
LL ans;
void dfs(int u,int fa)
{
    son[u] = 1;
    for(int i = head[u];i!=-1;i = es[i].next)
    {
        int v = es[i].to;
        LL w = es[i].w;
        if(v == fa) continue;
        dfs(v,u);
        son[u]+=son[v];
        LL t = min(son[v],k);
        ans+=t*w;
    }
}
int main()
{
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        e = 0; memset(head,-1,sizeof(head));
        memset(son,0,sizeof(son));
        int u,v;
        LL w;
        for(int i = 0;i < n-1;i++)
        {
            scanf("%d%d%lld",&u,&v,&w);
            addedge(u,v,w);
            addedge(v,u,w);
        }
        ans = 0;
        dfs(1,-1);
        printf("%lld\n",ans);
    }
    return 0;
}
/*
*/