hdu6060-思维&搜索&好题-RXD and dividing

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

给定一个树，问你把除了1节点之外的节点 分成k份，然后让每一份和1节点 连接成 最小的斯坦纳树（最小生成树是一种特殊的斯坦纳树，所以斯坦纳树就是把所有的节点连通的树。），而图中已经是树了，所以其实就是把他们连接起来（就是题中对应的f函数，但是已经是树了所以不存在最小一说。。连起来只有一种情况啊）。问你最大的花费是多少。

比赛的时候连题意都读错了，因为 k个集合包含了所有的点，所以必然有好多边要重复计算，我们如果想要最大的话，必须让其尽可能多的重复计算。于是就可以构建下图中的情况qwq



（图中假定k是3，若大于3，那么中间那两个点应该换成其他颜色，希望更多的和1节点邻接的边的重用 ！）

即尽可能的让其子树充满不同的颜色（集合）。取一个 k和siz的 最小值。

一次dfs就可以了。。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e6+12;
struct Node{
int to;
int cost;
Node(){};
Node(int _a,int _b){to=_a;cost=_b;}
};
vector<Node>G[maxn];
int siz[maxn];
int vc[maxn];
void dfs(int u,int fa){
//cout<<"!!!!"<<endl;
siz[u]=1;
for(int i=0;i<G[u].size();i++){
int v=G[u][i].to;
if(v==fa) continue;
dfs(v,u);
vc[v]=G[u][i].cost;
siz[u]+=siz[v];
}
//if(!flag) siz[u]=1;
return ;
}
int main()
{   int m,k;
int a,b,c;
while(~scanf("%d%d",&m,&k)){
for(int i=0;i<maxn;i++)
G[i].clear();
for(int i=0;i<m-1;i++){
scanf("%d%d%d",&a,&b,&c);
G[a].push_back(Node(b,c));
G[b].push_back(Node(a,c));
}
dfs(1,-1);
//puts("!!!");
ll all=0;
for(int i=2;i<=m;i++){
all+=1ll*min(siz[i],k)*vc[i];
}
printf("%lld\n",all);

}

return 0;
}