分成了未申请,申请成功和失败三种状态,转移就是类似的,不知道是我写挂了,还是这个做法不太行呢orz,大佬们求助,代码如下
#include<bits/stdc++.h>
#define to_l(a) ((a)<<1)
#define to_r(a) ((a)<<1|1)
#define lowbit(a) ((a)&(-a))
using namespace std;
typedef long long int ll;
typedef unsigned long long int ull;
const int int_inf=0x3f3f3f3f;
const ll ll_inf=0x3f3f3f3f3f3f3f3f;
const int max_n=2e3+5;
const int max_m=9e4+5;
double dp[2][max_n][3];
double p[max_n];
int ddd[305][305];
int c[max_n],d[max_n];
int dis(int a,int b)
{
return ddd[a][b];
}
void init(int n)
{
//memset(ddd,0x3f,sizeof(ddd));
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
for(int k=1;k<=n;k++){
ddd[j][k]=ddd[k][j]=min(ddd[j][k],ddd[j][i]+ddd[i][k]);
}
}
}
}
int main()
{
// ios::sync_with_stdio(false);
// cin.tie(0);
int n,m,v,e;
scanf("%d%d%d%d",&n,&m,&v,&e);
for(int i=1;i<=n;i++){
scanf("%d",c+i);
}
for(int i=1;i<=n;i++){
scanf("%d",d+i);
}
for(int i=1;i<=n;i++){
scanf("%lf",p+i);
}
for(int i=1;i<=v;i++){
for(int j=1;j<=v;j++){
ddd[i][j]=1e5;
}
}
for(int i=1;i<=e;i++){
int a,b,val;
scanf("%d%d%d",&a,&b,&val);
ddd[a][b]=min(ddd[a][b],val);
ddd[b][a]=min(ddd[b][a],val);
}
for(int i=1;i<=v;i++)
ddd[i][i]=0;
init(v);
for(int i=0;i<2;i++){
for(int j=0;j<=m;j++){
for(int k=0;k<3;k++){
dp[i][j][k]=1e18;
}
}
}
int k=0;
dp[!k][0][0]=0;
if(m!=0) dp[!k][1][1]=0,dp[!k][1][2]=0;
for(int i=2;i<=n;i++){
int t1=dis(d[i],c[i-1]);
int t2=dis(c[i],c[i-1]);
int t3=dis(c[i],d[i-1]);
int t4=dis(d[i],d[i-1]);
for(int j=0;j<=i;j++){
//dp[k][j][0]=dp[k][j][1]=dp[k][j][2]=1e18;
if(j-1>=0){
dp[k][j][2]=dp[!k][j-1][0]+t1;
if(j>1) dp[k][j][2]=min(dp[k][j][2],
(dp[!k][j-1][1]+t1)*(1-p[i-1])+(dp[!k][j-1][2]+t4)*(p[i-1]));
}
if(j-1>=0){
dp[k][j][1]=dp[!k][j-1][0]+t2;
if(j>1) dp[k][j][1]=min(dp[k][j][1],
(dp[!k][j-1][1]+t2)*(1-p[i-1])+(dp[!k][j-1][2]+t3)*(p[i-1]));
}
dp[k][j][0]=dp[!k][j][0]+t2;
if(j!=0) dp[k][j][0]=min(dp[k][j][0],
(dp[!k][j][1]+t2)*(1-p[i-1])+(dp[!k][j][2]+t3)*(p[i-1]));
}
k=!k;
}
double ans=1e18;
for(int i=0;i<=m;i++){
if(i==0){
ans=min(ans,dp[!k][i][0]);
}else{
ans=min(ans,min(dp[!k][i][0],dp[!k][i][1]*(1-p[n])+dp[!k][i][2]*(p[n])));
}
}
//cout<<ans<<endl;
printf("%.2lf\n",ans);
return 0;
}