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;}