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