Luogu P1850 [NOIp提高组2016]换教室

---

题目描述 传送门

---

作为蒟蒻的我不懂概率，不懂期望QWQ。看别人的题解裸地期望DP似懂非懂。先写一下吧，以后再做应该会更能理解的。

---

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int maxn=2010,maxv=305;
int a[maxn],b[maxn];
double c[maxn],f[maxn][maxn][2];
int d[maxv][maxv];
int main(){
    int n,m,v,e;
    cin>>n>>m>>v>>e;
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=n;i++) scanf("%d",&b[i]);
    for(int i=1;i<=n;i++) scanf("%lf",&c[i]);
    for(int i=1;i<=v;i++)
        for(int j=1;j<=v;j++)
            d[i][j]=1e9;
    for(int i=0;i<e;i++){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        if(a!=b&&c<d[a][b]) d[a][b]=d[b][a]=c;
    }
    for(int i=1;i<=v;i++) d[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(d[i][k]+d[k][j]<d[i][j]) d[i][j]=d[i][k]+d[k][j];
    for(int i=0;i<=n;i++)
        for(int j=0;j<=n;j++)
            f[i][j][0]=f[i][j][1]=1e9;
    f[1][0][0]=0;
    f[1][1][1]=0;
    for(int i=2;i<=n;i++){
        f[i][0][0]=f[i-1][0][0]+d[a[i-1]][a[i]];
        for(int j=1;j<=min(i,m);j++){
            f[i][j][0]=min(f[i-1][j][0]+d[a[i-1]][a[i]],f[i-1][j][1]+d[b[i-1]][a[i]]*c[i-1]+d[a[i-1]][a[i]]*(1-c[i-1]));
            f[i][j][1]=min(f[i-1][j-1][0]+d[a[i-1]][b[i]]*c[i]+d[a[i-1]][a[i]]*(1-c[i]),f[i-1][j-1][1]+d[b[i-1]][b[i]]*c[i]*c[i-1]+d[b[i-1]][a[i]]*c[i-1]*(1-c[i])+d[a[i-1]][b[i]]*(1-c[i-1])*c[i]+d[a[i-1]][a[i]]*(1-c[i])*(1-c[i-1]));
        }
    }
    double ans=f[n][0][0];
    for(int i=1;i<=m;i++)
        ans=min(ans,min(f[n][i][0],f[n][i][1]));
    printf("%.2lf",ans);
    return 0;
}