2017 多校训练第三场 HDU 6060 RXD and dividing

考场上被题面的斯坦纳树给吓到了。。。其实这个输入本身就是一个树，那么放在这个题里面就是找一个最小生成树。。。权当是为了理解吧

题目的意思是把2到n这些点分成k个部分，根据最小斯坦纳树的定义，(x,fax)这个边的贡献就是x字数内不同标号的个数diff(i),显然diff(i)小于等于min(sz(i),k).

通过构造让所有diff取得最大，答案就按照官方题解那样，虽然我也有点不明白。。。

#include <bits/stdc++.h>using namespace std;const int maxn=1e6+6;typedef long long ll;const ll mod=1e9+7;int n,k;typedef struct Node{int to,val;Node(){}Node(int x,int y):to(x),val(y){}}node;vector<node> edge[maxn];int sz[maxn],pre[maxn];inline void add_edge(int x,int y,int z){edge[x].push_back(node(y,z));edge[y].push_back(node(x,z));}void dfs(int u,int fa){sz[u]=1;for(int i=0;i<edge[u].size();i++){int v=edge[u][i].to;if(v==fa) continue;pre[v]=edge[u][i].val;dfs(v,u);sz[u]+=sz[v];}}int main(int argc, char const *argv[]){while(~scanf("%d %d",&n,&k)){int x,y,z;for(int i=1;i<=n;i++) edge[i].clear();for(int i=0;i<n-1;i++){scanf("%d %d %d",&x,&y,&z);add_edge(x,y,z);}dfs(1,-1);ll ans=0;for(int i=1;i<=n;i++){ans+=(ll)pre[i]*min(sz[i],k);}printf("%lld\n",ans);}return 0;}

∑x=2nw[x][fax]∗min(szx,k)\sum_{x = 2}^{n}{w[x][fa_x] * min(sz_x,