P1850 [NOIP2016]换教室，动态规划，概率dp

对于刚上大学的牛牛来说，他面临的第一个问题是如何根据实际情况申请合适的课程。

在可以选择的课程中，有 2n 节课程安排在 n 个时间段上。在第 i（1≤i≤n）个时间段上，两节内容相同的课程同时在不同的地点进行，其中，牛牛预先被安排在教室 c_i上课，而另一节课程在教室 d_i进行。

在不提交任何申请的情况下，学生们需要按时间段的顺序依次完成所有的 n 节安排好的课程。如果学生想更换第 ii节课程的教室，则需要提出申请。若申请通过，学生就可以在第 i 个时间段去教室 d_i 上课，否则仍然在教室 c_i 上课。

由于更换教室的需求太多，申请不一定能获得通过。通过计算，牛牛发现申请更换第 i 节课程的教室时，申请被通过的概率是一个已知的实数 k_i，并且对于不同课程的申请，被通过的概率是互相独立的。

学校规定，所有的申请只能在学期开始前一次性提交，并且每个人只能选择至多 m 节课程进行申请。这意味着牛牛必须一次性决定是否申请更换每节课的教室，而不能根据某些课程的申请结果来决定其他课程是否申请；牛牛可以申请自己最希望更换教室的 m 门课程，也可以不用完这 m 个申请的机会，甚至可以一门课程都不申请。

因为不同的课程可能会被安排在不同的教室进行，所以牛牛需要利用课间时间从一间教室赶到另一间教室。

牛牛所在的大学有 v 个教室，有 e 条道路。每条道路连接两间教室，并且是可以双向通行的。由于道路的长度和拥堵程度不同，通过不同的道路耗费的体力可能会有所不同。 当第 i（1≤i≤n−1）节课结束后，牛牛就会从这节课的教室出发，选择一条耗费体力最少的路径前往下一节课的教室。

现在牛牛想知道，申请哪几门课程可以使他因在教室间移动耗费的体力值的总和的期望值最小，请你帮他求出这个最小值。

https://www.luogu.org/problemnew/show/P1850

首先跑Floeyd求出每个教室之间的距离，即map数组。

状态：f[i][j][0/1]，表示前i个时间段，换了j次课，第三维表示换没换。

初始化：全部设为无穷大。

转移：

用c1=c[i-1],c2=c[i],d1=d[i-1],d2=d[i]分别表示上个时间段的上课教室、这个时间段的上课教室、上个时间段可以换到的上课教室、这个时间段可以换到的上课教室。

一、先考虑不换，第三维为0时：

1、上个时间段不换并且这个时间段也不换

2、上个时间段换了并且这个时间段不换

f[i][j][0]=min(f[i][j][0],f[i-1][j][0]+map[c1][c2]);
f[i][j][0]=min(f[i][j][0],f[i-1][j][1]+map[d1][c2]*k[i-1]+map[c1][c2]*(1-k[i-1]));

因为换成不成功是有概率的，这里乘上路程，即可求出期望。

二、再考虑换，第三维为1时：、

1、上个时间段不换并且这个时间段换了

2、上个时间段换了并且这个时间段也换（这里有好几种组合）

f[i][j][1]=min(f[i][j][1],f[i-1][j-1][0]+map[c1][d2]*k[i]+map[c1][c2]*(1-k[i]));
f[i][j][1]=min(f[i][j][1],f[i-1][j-1][1]+map[d1][d2]*k[i]*k[i-1]+map[d1][c2]*(1-k[i])*k[i-1]
+map[c1][d2]*k[i]*(1-k[i-1])+map[c1][c2]*(1-k[i])*(1-k[i-1]));

以上就是所有情况。

结果：因为不是必须换到m次，所以在所有f[n][1~m][1]与f[n][1~m][0]中取最小值即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#define inf 1e17+5
using namespace std;
int inline read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,m,v,e;
int map[301][301];
int c[2001],d[2001];
double k[2001],f[2001][2001][2];
int main()
{
    n=read();m=read();v=read();e=read();
    for(int i=1;i<=n;i++){
        c[i]=read();
    }
    for(int i=1;i<=n;i++){
        d[i]=read();
    }
    for(int i=1;i<=n;i++){
        scanf("%lf",&k[i]);
    }
    memset(map,63,sizeof(map));
    for(int i=1;i<=e;i++){
        int x=read(),y=read(),z=read();
        map[x][y]=map[y][x]=min(map[x][y],z);
    }
    for(int p=1;p<=v;p++){
        for(int i=1;i<=v;i++){
            for(int j=1;j<=v;j++){
                if(map[i][j]>map[i][p]+map[p][j]){
                    map[i][j]=map[i][p]+map[p][j];
                }
            }
        }
    }
    for(int i=1;i<=v;i++){
        map[i][i]=map[i][0]=map[0][i]=0;
    }
    for(int i=0;i<=n;i++){
        for(int j=0;j<=m;j++){
            f[i][j][0]=f[i][j][1]=inf;
        }
    }
    f[1][1][1]=0;f[1][0][0]=0;
    for(int i=2;i<=n;i++){
        f[i][0][0]=f[i-1][0][0]+map[c[i-1]][c[i]];
        for(int j=1;j<=min(m,i);j++){
            int c1=c[i-1],c2=c[i],d1=d[i-1],d2=d[i];
            f[i][j][0]=min(f[i][j][0],f[i-1][j][0]+map[c1][c2]);
            f[i][j][0]=min(f[i][j][0],f[i-1][j][1]+map[d1][c2]*k[i-1]+map[c1][c2]*(1-k[i-1]));
            f[i][j][1]=min(f[i][j][1],f[i-1][j-1][0]+map[c1][d2]*k[i]+map[c1][c2]*(1-k[i]));
            f[i][j][1]=min(f[i][j][1],f[i-1][j-1][1]+map[d1][d2]*k[i]*k[i-1]+map[d1][c2]*(1-k[i])*k[i-1]
            +map[c1][d2]*k[i]*(1-k[i-1])+map[c1][c2]*(1-k[i])*(1-k[i-1]));
        }
    }
    double ans=inf;
    for(int i=0;i<=m;i++){
        ans=min(ans,min(f[n][i][1],f[n][i][0]));
    }
    printf("%.2lf",ans);
    return 0;
}