HDU 6060 RXD and dividing（思维+计算贡献值）

http://acm.hdu.edu.cn/showproblem.php?pid=6060

题意：

给定一棵 n 个节点的树，1 为根。现要将节点 2 ~ n 划分为 k 块，使得每一块与根节点形成的最小斯坦纳树的边权值总和最大。

思路：

这道题目应该往边的贡献值方向去思考，对于u（非1）结点，设它与它父亲结点的边为e，它的子节点分的块越多，那么e这条边的贡献值也就越大，因为每一分块都需要e这条边来连通。所以我们就要尽量让每条边的贡献值都最大。

下面图解一下：（分成 3 part的情况）

4、5、6、7由于超过了3，所以只能分成3部分，我们可以假设性的分为{4,5}，{6}，{7}这样3组，1和3是没有子树关系的，所以可以把它们合并进去，不会影响3结点边的贡献值，

现在又可以假设性的分为{4,5,8}，{6,9}，{7,10}这样三组，将2结点合并进去后可以变成{4,5,8}，{2,6,9}，{7,10}这样三组，此时1、2、3、4、5、6、7、8、9这几个结点的边都已经达到了它们所能达到的最大贡献值。

依次向上分析即可。

所以，我们可以得出结论，每条边的最大贡献值就是min（sz[u]，k），最后将所有边的贡献值加起来即可。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<sstream>
#include<vector>
#include<stack>
#include<queue>
#include<cmath>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
typedef pair<int,int> pll;
const int INF = 0x3f3f3f3f;
const int maxn = 1000000 + 5;

int n, k;

int sz[maxn];
int val[maxn];
vector<pll> G[maxn];

void dfs(int u, int fa)
{
    sz[u]=1;
    for(int i=0;i<G[u].size();i++)
    {
        int v=G[u][i].first;
        if(v==fa)  continue;
        val[v]=G[u][i].second;
        dfs(v,u);
        sz[u]+=sz[v];
    }
}

int main()
{
    //freopen("in.txt","r",stdin);
    while(~scanf("%d%d",&n,&k))
    {
        for(int i=1;i<=n;i++)  G[i].clear();

        for(int i=1;i<n;i++)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            G[u].push_back(make_pair(v,w));
            G[v].push_back(make_pair(u,w));
        }

        dfs(1,-1);
        ll ans=0;
        for(int i=1;i<=n;i++)
            ans+=(ll)val[i]*min(k,sz[i]);
        printf("%lld
",ans);
    }
    return 0;
}