题目大意:

给你 n 个点和 n-1 条边组成一颗树,然后让你把这 n 个点分成 k 组,每一组的值为这一组所有点和 1 号
结点组成的最小斯坦纳树的边权之和。现在问你如何划分才能得到使得这k组的值最大。

分析:

首先我们考虑一件事,那就是,对于任意一条边,无论如何划分,这条边只有可能最多被选取 k 次。
现在我来定义一种选取方法。对于一条边,它能被选取的最多次数为:min{左边的点的个数,右边的点的个数,k}。那我根据这种办法我就可以得到每条边最多能被选取的次数。这样我现在只要证明一定存在一种划分方式使得每条边都取到最多的选取次数,那么就可以以 O(n) 的时间复杂度求得这 k 组划分的最大值。

下面开始证明:

首先对于最多选取次数小于 k 的边,我只要让靠近叶子方向的结点和它的子孙节点分在不用的组就可以了。
然后再考虑最多选取次数等于 k 的边,那我把这些点随便某些分到一组,只要是k组(每组不空),那么都可以使得该边一共被选取k次。也是可以在不影响子节点划分方式的情况下把该边靠近叶子结点的结点随便分到哪一个组里面去来得到满足条件的划分方式。
证毕~

代码:

#include<bits/stdc++.h>
#define MAXN 1000050
using namespace std;
struct EDGE
{
    int e,v;
};

//vector<EDGE>edge;
vector<EDGE>edge[MAXN];
int n,k;
long long int ans=0;
bool vis[MAXN];

void Add_edge(int s , int e , int v)
{
    EDGE l;     l.e=e;      l.v=v;
    edge[s].push_back(l);
}

int dfs(int x)
{
    vis[x]=1;
    if(edge[x].size()==1&&x!=1)return 1;
    int num=0;
    for(int i=0;i<edge[x].size();i++)
    {
        if(vis[edge[x][i].e]==1)continue;
        int t=dfs(edge[x][i].e);
        num+=t;
        ans+=(long long int)t*(long long int )edge[x][i].v;
    }
    return  (num+1>k)?k:(num+1) ;
}

int main()
{
    //feropen("in.txt","r",std);
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        memset(vis,0,sizeof(vis));
        ans=0;
        for(int i=1;i<n;i++)
        {
            int ls,le,lv;
            scanf("%d%d%d",&ls,&le,&lv);
            Add_edge(ls,le,lv);
            Add_edge(le,ls,lv);
        }
        dfs(1);
        printf("%lld\n",ans);
        for(int i=0;i<=n;i++)edge[i].clear();
    }

}