题意:虽然有中文解释,不过还是说下,给一个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;
}