洛谷1850(NOIp2016) 换教室——期望dp

题目：https://www.luogu.org/problemnew/show/P1850

状态里记录的是”上一回有没有申请“，而不是”上一回申请成功否“，不然“申请 j 次”就没法转移了。

double不能memset，所以手动。

别忘了dis[ i ][ i ]=0。

有重边！！！所以读入边的时候取一下min！

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define db double
using namespace std;
const int N=2005,V=305,M=90005;
const db INF=0x3f3f3f3f;
int n,K,num,ent,c[N],d[N],dis[V][V];
db p[N],dp[2][N][2],ans=INF;
void mmst(db a[N][2])
{
    for(int j=0;j<=K;j++)a[j][0]=a[j][1]=INF;
}
int main()
{
    scanf("%d%d%d%d",&n,&K,&num,&ent);
    int x,y,z;
    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",&p[i]);
    memset(dis,0x3f,sizeof dis);
    while(ent--)
    {
        scanf("%d%d%d",&x,&y,&z);
        dis[x][y]=min(dis[x][y],z);dis[y][x]=dis[x][y];//?!////////
//        dis[x][y]=dis[y][x]=z;
    }
    for(int i=1;i<=num;i++)dis[i][i]=0;//
    for(int k=1;k<=num;k++)
        for(int i=1;i<=num;i++)
            for(int j=1;j<=num;j++)
                dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]);
    mmst(dp[1]);
    dp[1][0][0]=dp[1][1][1]=0;
    for(int i=2;i<=n;i++)
    {
        int x=c[i-1],y=c[i],u=d[i-1],v=d[i],fx=(i&1);
        mmst(dp[fx]);
        for(int j=0;j<=K;j++)
        {
            dp[fx][j][0]=min(dp[!fx][j][0]+dis[x][y]
                ,dp[!fx][j][1]+p[i-1]*dis[u][y]+(1-p[i-1])*dis[x][y]);
            if(j)dp[fx][j][1]=min(dp[!fx][j-1][0]+p[i]*dis[x][v]+(1-p[i])*dis[x][y]
                ,dp[!fx][j-1][1]+p[i]*p[i-1]*dis[u][v]+p[i]*(1-p[i-1])*dis[x][v]
                            +(1-p[i])*p[i-1]*dis[u][y]+(1-p[i])*(1-p[i-1])*dis[x][y]);
        }
    }
    int fx=(n&1);
    for(int j=0;j<=K;j++)ans=min(ans,min(dp[fx][j][0],dp[fx][j][1]));
    printf("%.2lf
",ans);
    return 0;
}