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,