80pts求助
查看原帖
80pts求助
222057
joy2010WonderMaker楼主2022/10/15 20:30
#include<bits/stdc++.h>
using namespace std;
int n,m,v,e,c[2005],d[2005],a,b,g[305][305];
double k[2005],dp[2005][2005][2],w,ans=1e17+7,dis[305][305]; 
int main(){
	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",k+i);
	for(int i=1;i<=v;i++)
		for(int j=1;j<=v;j++)
			dis[i][j]=1e9+7;
	for(int i=1;i<=e;i++){
		scanf("%d%d%lf",&a,&b,&w);
		dis[a][b]=dis[b][a]=min(dis[a][b],w);
	}
	for(int i=1;i<=v;i++)
		dis[i][i]=0;
	for(int k=1;k<=v;k++)
		for(int i=1;i<=v;i++)
			for(int j=1;j<=v;j++)
				dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]);
	for(int i=1;i<=n;i++)
		for(int j=0;j<=m;j++)
			dp[i][j][0]=dp[i][j][1]=1e15+7; 
	dp[1][0][0]=dp[1][1][1]=0;
	for(int i=2;i<=n;i++){
		dp[i][0][0]=dp[i-1][0][0]+dis[c[i-1]][c[i]];
		for(int j=1;j<=m;j++){
			dp[i][j][0]=min({dp[i][j][0],dp[i-1][j][0]+dis[c[i-1]][c[i]],dp[i-1][j][1]+k[i-1]*dis[d[i-1]][c[i]]+(1-k[i-1])*dis[c[i-1]][c[i]]});
			dp[i][j][1]=min({dp[i][j][1],dp[i-1][j-1][0]+k[i]*dis[c[i-1]][d[i]]+(1-k[i])*dis[c[i-1]][c[i]],dp[i-1][j-1][1]+k[i-1]*k[i]*dis[d[i-1]][d[i]]+(1-k[i-1])*k[i]*dis[c[i-1]][d[i]]+k[i-1]*(1-k[i])*dis[d[i-1]][c[i]]+(1-k[i-1])*(1-k[i])*dis[d[i-1]][d[i]]});
		}
	}
	for(int i=0;i<=m;i++)
		ans=min({ans,dp[n][i][0],dp[n][i][1]});
	printf("%.2lf",ans);
	return 0;
}

WA on 2,16,17,18,23

2022/10/15 20:30
加载中...