hdu 6060 RXD and dividing （dfs）

怎么说呢，比赛的时候没看懂题，对就是没看懂题，应该多读几遍题，不要浮躁，浮躁解决不了问题，求得是最大的res，f（s）是最小的，就是求1-s的最短路径吧，说的应该是唯一的那条路径，不要走1-2-1-2-3直接1-2-3；然后求res最大。

把1看成整棵树的根.
问题相当于把2∼n2\sim
n2∼n每个点一个[1,k][1,
k][1,k]的标号.
然后根据最小斯坦纳树的定义, (x,fax)(x,
fa_x)(x,fa​x​​) 这条边的贡献是
x 子树内不同标号的个数目difidif_idif​i​​.
那么显然有difi≤min(k,szi)dif_i\leq
min(k, sz_i)dif​i​​≤min(k,sz​i​​), szisz_isz​i​​表示子树大小.
可以通过构造让所有difidif_idif​i​​都取到最大值.
所以答案就是∑x=2nw[x][fax]∗min(szx,k)\sum_{x
= 2}^{n}{w[x][fa_x] * min(sz_x, k)}∑​x=2​n​​w[x][fa​x​​]∗min(sz​x​​,k)时间复杂度O(n)O(n)O(n).

fax就是x的父节点。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1000000+10;
struct node
{
int v,w;
node(){}
node(int _v,int _w)
{
v=_v;
w=_w;
}
};
vector<node>map1[maxn];
int pre[maxn],cnt[maxn];

void dfs(int u,int fa)
{
cnt[u]=1;
for(int i=0;i<map1[u].size();i++)
{
int v=map1[u][i].v;
if(v==fa)continue;
pre[v]=map1[u][i].w;
dfs(v,u);
cnt[u]+=cnt[v];
}
}
int main()
{
int n,k;
while(~scanf("%d%d",&n,&k))
{
int u,v,w;
for(int i=1;i<n;i++)
{
scanf("%d%d%d",&u,&v,&w);
map1[u].push_back(node(v,w));
map1[v].push_back(node(u,w));
}
dfs(1,-1);
long long ans=0;
for(int i=2;i<=n;i++)
{
ans+=(long long) pre[i]*min(k,cnt[i]);
}
printf("%lld\n",ans);
for(int i=0;i<=n;i++)map1[i].clear();
}
return 0;
}