NOIP2016 换教室

传送门
其实这个题多重构几次代码就过了。 ——来自hzy dalao的吐槽
我也不多说什么，期望dp没什么可说的，主要是这个题比较基础，就是实现坑，最好自己踩一踩坑。
代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstdlib>
#define ll long long
using namespace std;
inline int read(){
    int x=0;char ch=' ';int f=1;
    while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
    if(ch=='-')f=-1,ch=getchar();
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
const int N=305,M=2e3+5;
int n,m,v,e;
int g[N][N],c[M],d[M];
double k[M],dp[2][M][M];
int main(){
    n=read();m=read();v=read();e=read();
    for(int i=1;i<=v;i++)for(int j=1;j<=v;j++)g[i][j]=0x3f3f3f3f;
    for(int i=1;i<=v;i++)g[i][i]=0;
    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]);
    for(int i=1;i<=e;i++){
        int x=read(),y=read(),l=read();
        g[x][y]=g[y][x]=min(g[x][y],l);
    }
    for(int K=1;K<=v;K++){
        for(int i=1;i<=v;i++){
            for(int j=1;j<=v;j++){
                g[i][j]=min(g[i][j],g[i][K]+g[K][j]);
            }
        }
    }
    for(int i=1;i<=n;i++)for(int j=0;j<=m;j++)dp[0][i][j]=dp[1][i][j]=0x3f3f3f3f;
    dp[0][1][0]=0;dp[1][1][1]=0;
    for(int i=2;i<=n;i++){
        for(int j=0;j<=min(i,m);j++){
            dp[0][i][j]=min(dp[0][i-1][j]+(double)g[c[i-1]][c[i]],
                            dp[1][i-1][j]+(double)g[c[i-1]][c[i]]*(1.0-k[i-1])
                            +(double)g[d[i-1]][c[i]]*k[i-1]);
            if(j>0){
                dp[1][i][j]=min(dp[0][i-1][j-1]
                                +(double)g[c[i-1]][c[i]]*(1.0-k[i])
                                +(double)g[c[i-1]][d[i]]*k[i],
                                dp[1][i-1][j-1]
                                +(double)g[c[i-1]][c[i]]*(1.0-k[i])*(1.0-k[i-1])
                                +(double)g[c[i-1]][d[i]]*k[i]*(1.0-k[i-1])
                                +(double)g[d[i-1]][c[i]]*(1.0-k[i])*k[i-1]
                                +(double)g[d[i-1]][d[i]]*k[i]*k[i-1]);
            }
        }
    }
    double ans=0x3f3f3f3f;
    for(int i=0;i<=m;i++)ans=min(ans,min(dp[0][n][i],dp[1][n][i]));
    printf("%.2lf",ans);
    return 0;
}