noip2016 换教室 期望+DP

考场一看见期望就怂了，于是去做第二题，做了两个多小时，没做出来。GG。
dp[ i ][ j ][0/1]表示前 i 节课申请 j 次的最小期望。然后[0]表示这次没申请，[1]表示已申请。
所以dp[i-1][j][1] 有f[i-1]的概率停在 d[i-1] 点，有（1 - f[i-1]）概率停在c[i-1]点。
dp[i][j][0]=min
dp[i-1][j][1] +dis[ d[i-1] ][ c[i] ] * f[i-1] + dis[ c[i-1] ][ c[i] ]*( 1-f[i-1]）
dp[i-1][j][0] + dis[c[i-1]][c[i]]
dp[i][j][1]转移同理。
这里的第一维应该不用也可以，为了考虑方便就加上了，反正不卡空间。。。

#include<iostream>
#include<cstdio>
#define maxn 2005
using namespace std;
int c[maxn],d[maxn];
double f[maxn];
int dis[305][305];
int inf=1e9;
double dp[maxn][maxn][2];
int main()
{
    int n,m,V,e;
    scanf("%d%d%d%d",&n,&m,&V,&e);
    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",&f[i]);
    int u,v,w;
    for(int i=1;i<=V;i++)
    for(int j=1;j<=V;j++)
        dis[i][j]=inf;
    for(int i=1;i<=e;i++)
    {
        scanf("%d%d%d",&u,&v,&w);
        dis[u][v]=min(dis[u][v],w);
        dis[v][u]=min(dis[v][u],w);
    }
    for(int i=1;i<=V;i++)
        dis[i][i]=0;
    for(int k=1;k<=V;k++)
    for(int i=1;i<=V;i++)
    for(int j=1;j<=V;j++)
        if(i!=j&&i!=k&&j!=k)
        dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]);
    for(int i=0;i<=n;i++)
    for(int j=0;j<=m;j++)
        dp[i][j][0]=dp[i][j][1]=inf;
    dp[1][0][0]=0;dp[1][1][0]=0;dp[1][1][1]=0;
    for(int i=2;i<=n;i++)
        dp[i][0][0]=dp[i-1][0][0]+dis[c[i-1]][c[i]];
    for(int i=2;i<=n;i++)
    for(int j=1;j<=min(m,i);j++)
    {
        double p=dp[i-1][j][1]+dis[d[i-1]][c[i]]*f[i-1]+dis[c[i-1]][c[i]]*(1.0-f[i-1]);
        double q=dp[i-1][j][0]+dis[c[i-1]][c[i]];
        dp[i][j][0]=min(p,q);
        p=dp[i-1][j-1][0]+dis[c[i-1]][d[i]]*f[i]+dis[c[i-1]][c[i]]*(1-f[i]);
        q=dp[i-1][j-1][1]+f[i-1]*(dis[d[i-1]][c[i]]*(1.0-f[i])+dis[d[i-1]][d[i]]*f[i]);
        q+=(1.0-f[i-1])*(dis[c[i-1]][c[i]]*(1.0-f[i])+dis[c[i-1]][d[i]]*f[i]);
        dp[i][j][1]=min(p,q);
    }
    double ans=1e9;
    for(int i=0;i<=m;i++)
        ans=min(ans,min(dp[n][i][0],dp[n][i][1]));
    printf("%.2lf",ans);
    return 0;
}