1.dp用以处理相邻间有递推关系的(可以复杂多维)如这里第i个的选择只会影响前一个和后一个
2.记录状态,三维dp[i][j][k]代表当前是第i个,j表示前i-1个有多少个已经选择,k表示当前是否选择
这里第二维的范围是限制,跑的时候可以跑大一些,最后更新ans可以按题目限定的范围内跑就可以
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[305][305];
int c[2005],d[2005];
int n,m,vv,e;
double p[2005];
double dp[2005][2005][2];
void floyd()
{
for(int k=1;k<=vv;++k)
for(int i=1;i<=vv;++i)
for(int j=1;j<=vv;++j)
{
a[i][j]=min(a[i][j],a[i][k]+a[k][j]);
}
}
int main()
{
scanf("%d%d%d%d",&n,&m,&vv,&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]);
}
int u,v,w;
memset(a,0x3f3f3f3f,sizeof(a));
for(int i=1;i<=vv;++i)
a[i][i]=0;
for(int i=1;i<=e;++i)
{
scanf("%d%d%d",&u,&v,&w);
a[u][v]=a[v][u]=min(a[u][v],w);
}
floyd();
memset(dp,0x3f3f3f3f,sizeof(dp));
dp[1][0][0]=0;
dp[1][0][1]=0;
double k11;
for(int i=2;i<=n;++i)
{
dp[i][0][0]=dp[i-1][0][0]+a[c[i]][c[i-1]];
dp[i][0][1]=dp[i-1][0][0]+p[i]*(a[c[i-1]][d[i]])+(1-p[i])*(a[c[i]][c[i-1]]);
for(int j=1;(j<=n);++j)
{
k11=p[i-1]*p[i]*(a[d[i-1]][d[i]])+(1-p[i-1])*p[i]*a[c[i-1]][d[i]]+(1-p[i-1])*(1-p[i])*a[c[i]][c[i-1]]+(1-p[i])*p[i-1]*a[d[i-1]][c[i]];
dp[i][j][0]=min(dp[i-1][j][0]+a[c[i]][c[i-1]],dp[i-1][j-1][1]+p[i-1]*(a[d[i-1]][c[i]])+(1-p[i-1])*a[c[i-1]][c[i]]);
dp[i][j][1]=min(dp[i-1][j][0]+p[i]*(a[d[i]][c[i-1]])+(1-p[i])*(a[c[i]][c[i-1]]),dp[i-1][j-1][1]+k11);
}
}
double ans=dp[n][0][0];
if(m==0)
{
ans*=100;
ans=round(ans);
ans/=100;
printf("%.2lf",ans);
return 0;
}
ans=min(ans,dp[n][0][1]);
for(int i=1;(i<n)&&(i<m);++i)
{
ans=min(ans,dp[n][i][0]);
ans=min(ans,dp[n][i][1]);
}
if(m<=n)
{
ans=min(ans,dp[n][m][0]);
}
ans*=100;
ans=round(ans);
ans/=100;
printf("%.2lf",ans);
return 0;
}