BestCoder Round #23 1003 Cities

题意：虽然有中文解释，不过还是说下，给一个n个节点的树，然后找到k个点，让这k个节点距离的期望最小。这里的期望是这么算的，从k个点中随机选取两个点（可以是一个点），然后就是距离的和除以k*k，最终结果要求乘上k*k，也就是这个子树的任意两点的距离和

思路：原来做过一个类似的，不过是求某棵树上任意两点的距离和，这两个题的思路是类似的。都是dfs，对于u->v这个边，经过的次数就是两侧的节点数a*b*2*权值，这个题也是这么搞，不过在用子节点更新父节点的值的时候需要注意是01背包，也就是需要从大到小枚举。

还有一个错的比较多的地方就是dp[u][j] = min(dp[u][j],dp[u][j-t]+dp[v][t]+side[i].w*(k-t)*t*2);，是t*（k-t），之所以是k-t是因为算的是这条边经过的次数

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#define ll long long
using namespace std;
const int maxn = 2010;
const int maxm = 6020;
const ll INF = 1000000000000LL;
ll dp[maxn][60];
int n,k;
struct Side{
    int v;
    ll w;
    int next;
}side[maxm];
int top;
int node[maxn];
void add_side(int u,int v,ll w){
    side[top] = (Side){v,w,node[u]};
    node[u] = top++;
}
void dfs(int u,int pre){
    dp[u][1] = 0;
    for(int i = node[u];i != -1;i = side[i].next){
        int v = side[i].v;
        if(v == pre)continue;
        dfs(v,u);
        for(int j = k;j >= 1;j --){
            for(int t = j-1;t >= 1;t --){
                if(dp[u][j-t] == INF||dp[v][t] == INF)continue;
                dp[u][j] = min(dp[u][j],dp[u][j-t]+dp[v][t]+side[i].w*(k-t)*t*2);
            }
        }
    }
}
int main(){
    int T;
    scanf("%d",&T);
    while(T --){
        scanf("%d%d",&n,&k);
        top = 0;
        memset(node,-1,sizeof(node));
        for(int i = 0;i < n-1;i ++){
            int a,b;
            ll w;
            scanf("%d%d%I64d",&a,&b,&w);
            add_side(a,b,w);
            add_side(b,a,w);
        }
        for(int i = 0;i <= n;i ++){
            for(int j = 0;j <= k;j ++){
                dp[i][j] = INF;
            }
        }
        dfs(1,-1);
        ll ans = INF;
        for(int i = 1;i <= n;i ++){
            ans = min(ans,dp[i][k]);
        }
        printf("%I64d\n",ans);
    }
    return 0;
}