洛谷 P1850 换教室 （Floyd + 期望dp）

题意

共有n节课，每节课有两间教室，分为初始教室和可申请教室，拥有m次申请换教室的机会，每次申请换教室时有ki的概率换成功，默认在初始教室上课。总共v间教室，教室间共有e条路径权值为w，求在m次机会内如何更换，使得总权值期望最小。

题解

首先对所有教室求一次Floyd，得到教室间最短路。
求dp[i][j][k]
其中i表示第i节课，j表示算上这一次一共申请了j次，k表示这节课是否申请。

dp[i][j][0]

若上一次不换，则dp[i][j][0]=dp[i][j][0]+上一次不换的教室到这一次不换的教室权值
若上一次换，则dp[i][j][0]=dp[i][j][1]+上一次换成功的概率乘以权值+上一次换失败的概率乘以权值

dp[i][j][1]

若上一次不换，则dp[i][j][1]=dp[i][j-1][0]+这一次换成功概率乘以权值+这一次换失败概率乘以权值
若上一次换，同理，这里概率需要算上一次的和这一次的，需要算4个

反思

Floyd写残了，跑的顺序错了，导致一直卡在52分。

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<string.h>
#include<set>
#include<queue>
#include<stack>
#include<math.h>
#include<map>
using namespace std;

typedef long long int ll;
#define lson rt<<1
#define rson rt<<1|1
const ll mod=100000000;
const ll maxn=2020;
const double inf=0x7f7f7f;

double dp[maxn][maxn][3];
double g[320][320];
int C0[maxn];
int C1[maxn];
double k[maxn];
int n,m,v,e;
void init()
{
    for(int i=1;i<=n;i++)
        for(int j=0;j<=m;j++)
            dp[i][j][0]=dp[i][j][1]=999999999;
    dp[1][0][0]=0;
    dp[1][1][1]=0;
    //dp[1][1][0]=0;
    for(int i=0;i<=310;i++)
    {
        for(int j=0;j<=310;j++)
        {
            g[i][j]=999999999;
        }
        g[i][i]=0;

    }
}



int main(){

    cin>>n>>m>>v>>e;
    init();
    for(int i=1;i<=n;i++)
    {
        cin>>C0[i];
    }
    for(int i=1;i<=n;i++)
    {
        cin>>C1[i];
    }
    for(int i=1;i<=n;i++)
    {
        cin>>k[i];
    }
    for(int i=1;i<=e;i++)
    {
        int a,b;
        double w;
        scanf("%d%d%lf",&a,&b,&w);
        g[a][b]=min(g[a][b],w);
        g[b][a]=min(g[b][a],w);
    }
    for(int k=1;k<=v;k++)
        for(int i=1;i<=v;i++)
        {
            for(int j=1;j<i;j++)
            {

                    if(g[i][j]>g[i][k]+g[k][j])
                        g[i][j]=g[j][i]=g[i][k]+g[k][j];

            }
        }

    for(int i=2;i<=n;i++)
    {
        for(int j=0;j<=n&&j<=m;j++)
        {

            dp[i][j][0]=min(dp[i-1][j][0]+g[C0[i-1]][C0[i]],
                            dp[i-1][j][1]+
                            k[i-1]*g[C1[i-1]][C0[i]]+
                            (1-k[i-1])*g[C0[i-1]][C0[i]]);

            if(j>=1)
                dp[i][j][1]=min(dp[i-1][j-1][0]+
                                k[i]*g[C0[i-1]][C1[i]]+(1-k[i])*g[C0[i-1]][C0[i]],
                                dp[i-1][j-1][1]+
                                k[i-1]*k[i]*g[C1[i-1]][C1[i]]+
                                k[i-1]*(1-k[i])*g[C1[i-1]][C0[i]]+
                                (1-k[i-1])*k[i]*g[C0[i-1]][C1[i]]+
                                (1-k[i-1])*(1-k[i])*g[C0[i-1]][C0[i]]);
        }
    }

    double ans=999999999;
    for(int i=0;i<=m;i++)
    {
        for(int k=0;k<=1;k++)
            ans=min(ans,dp[n][i][k]);

    }
    printf("%.2lf\n",ans);

}