做这个题要先搞懂期望。。。
我还是不懂什么是期望。。。
但好像这个题的期望是距离乘概率。。。
所以这就是个期望dp
f[i][j][0,1] 表示到i时间段,选了j个教室,选不选当前的教室
所以
有四种情况,前面选后面不选,前面选后面选,前面不选后面选,前面不选后面不选
方程就是
f[i][j][0]=min(f[i-1][j][0]+dis[c[i-1]][c[i]],f[i-1][j][1]+k[i-1]*dis[d[i-1]][c[i]]+(1-k[i-1])*dis[c[i-1]][c[i]]);
if(j>0) f[i][j][1]=min(f[i-1][j-1][0]+k[i]*dis[c[i-1]][d[i]]+(1-k[i])*dis[c[i-1]][c[i]],
f[i-1][j-1][1]+k[i-1]k[i]*dis[d[i-1]][d[i]]+k[i-1](1-k[i])dis[d[i-1]][c[i]]+(1-k[i-1])*k[i]*dis[c[i-1]][d[i]]+(1-k[i-1])(1-k[i])*dis[c[i-1]][c[i]]);
然后找最小的
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;
int n,m,v,e;
int c[2001],d[2001];int dis[301][301];
double f[2001][2001][2],k[2001],ans=1e9+7;
int main(){
freopen("a.in","r",stdin);
scanf("%d%d%d%d",&n,&m,&v,&e);memset(dis,127/3,sizeof dis);
for(int i=1;i<=n;i++) scanf("%d",&c[i]);
for(int i=1;i<=n;i++) scanf("%d",&d[i]);
for(int i=1;i<=n;i++) scanf("%lf",&k[i]);
while(e--){
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
dis[x][y]=dis[y][x]=min(dis[x][y],z);
}
memset(f,127,sizeof f);f[1][0][0]=0;f[1][1][1]=0;
for(int q=1;q<=v;q++)
for(int i=1;i<=v;i++)
for(int j=1;j<=v;j++)
if(i!=j&&i!=q&&q!=j)
dis[i][j]=min(dis[i][j],dis[i][q]+dis[q][j]);
for(int i=1;i<=v;i++) dis[i][i]=0;
for(int i=2;i<=n;i++)
for(int j=0;j<=min(i,m);j++){
f[i][j][0]=min(f[i-1][j][0]+dis[c[i-1]][c[i]],f[i-1][j][1]+k[i-1]*dis[d[i-1]][c[i]]+(1-k[i-1])*dis[c[i-1]][c[i]]);
if(j>0)
f[i][j][1]=min(f[i-1][j-1][0]+k[i]*dis[c[i-1]][d[i]]+(1-k[i])*dis[c[i-1]][c[i]],
f[i-1][j-1][1]+k[i-1]*k[i]*dis[d[i-1]][d[i]]+k[i-1]*(1-k[i])*dis[d[i-1]][c[i]]+(1-k[i-1])*k[i]*dis[c[i-1]][d[i]]+(1-k[i-1])*(1-k[i])*dis[c[i-1]][c[i]]);
}
for(int i=0;i<=m;i++)
ans=min(ans,min(f[n][i][0],f[n][i][1]));
printf("%.2f",ans);
}