洛谷P1850 [noip2016]换教室——期望DP

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

注释掉了一堆愚蠢，自己还是太嫩了...

首先要注意选或不选是取 min 而不是 /2 ，因为这里的选或不选不是随机的而是取最优的；

然后注意 double 类型不能 memset ，还是要手动赋值；

其实也不算很难，却调了很久...

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int const maxn=;
int n,m,v,e,dis[][],c[maxn],d[maxn];
double p[maxn],f[maxn][maxn][],ans;
int rd()
{
    int ret=,f=;char ch=getchar();
    while(ch<''||ch>''){if(ch=='-')f=-; ch=getchar();}
    while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
    return ret*f;
}
int main()
{
    n=rd(); m=rd(); v=rd(); e=rd();
    for(int i=;i<=n;i++)c[i]=rd();
    for(int i=;i<=n;i++)d[i]=rd();
    for(int i=;i<=n;i++)scanf("%lf",&p[i]);
    memset(dis,0x3f,sizeof dis);
    for(int i=,x,y,z;i<=e;i++)
    {
        x=rd(); y=rd(); z=rd();
        dis[x][y]=min(dis[x][y],z); dis[y][x]=dis[x][y];//
    }
    for(int i=;i<=v;i++)dis[i][i]=;//!!!
    for(int k=;k<=v;k++)
        for(int i=;i<=v;i++)
            for(int j=;j<=v;j++)
                dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]);
//    memset(f,1,sizeof f);//!!!
//    for(int j=0;j<=m;j++)
//    {
//        f[n][j][0]=0; f[n][j][1]=0;
//    }
//    for(int i=n-1;i;i--)
//        for(int j=0;j<=m;j++)
//        {
//            f[i][j][0]=(f[i+1][j][0]+f[i+1][j][1])/2+(p[i-1]*dis[d[i]][c[i+1]]+(1-p[i-1])*dis[c[i]][c[i+1]]);
//            if(j)f[i][j][1]=(f[i+1][j-1][0]+f[i+1][j-1][1])/2+
//                        p[i]*(p[i-1]*dis[d[i]][d[i+1]] + (1-p[i-1])*dis[c[i]][d[i+1]])/2
//                        +(1-p[i])*dis[c[i]][c[i+1]]/2;
//        }
//    printf("%lf",f[1][m][0]+f[1][m][1]);
    for(int i=;i<=n;i++)//手动memset
        for(int j=;j<=m;j++)
            for(int k=;k<=;k++)f[i][j][k]=0x3f3f3f3f;
    f[][][]=; f[][][]=;
    for(int i=;i<=n;i++)
        for(int j=;j<=m&&j<=i;j++)
        {
            f[i][j][]=min(f[i-][j][] + dis[c[i-]][c[i]],
                           f[i-][j][] + (-p[i-])*dis[c[i-]][c[i]] + p[i-]*dis[d[i-]][c[i]]);
            if(j)
            f[i][j][]=min(f[i-][j-][] + (-p[i])*dis[c[i-]][c[i]] + p[i]*dis[c[i-]][d[i]],
                           f[i-][j-][] + (-p[i-])*(-p[i])*dis[c[i-]][c[i]] + p[i-]*p[i]*dis[d[i-]][d[i]] +
                                               (-p[i-])*p[i]*dis[c[i-]][d[i]] + p[i-]*(-p[i])*dis[d[i-]][c[i]]);
        }
    ans=0x3f3f3f3f;
    for(int j=;j<=m;j++)
        ans=min(ans,min(f[n][j][],f[n][j][]));
    printf("%.2lf",ans);
    return ;
}