传送门
其实这个题多重构几次代码就过了。 ——来自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;
}