2016noip提高组day1 换教室

做这个题要先搞懂期望。。。
我还是不懂什么是期望。。。
但好像这个题的期望是距离乘概率。。。
所以这就是个期望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);
}