求助76pts
查看原帖
求助76pts
723198
AAA404楼主2023/8/8 20:37

rt,讨论区看了,76pts的问题都没有

#include<bits/stdc++.h>
using namespace std;
const int N=2005;
int n,m,v,e,c[N],d[N];
double dis[N][N],f[N][N][2],p[N];
int main()
{
#ifdef LOCAL
 	freopen("1.in","r",stdin);
 	freopen("1.out","w",stdout);
#endif
    ios::sync_with_stdio(0);
 	cin.tie(0);cout.tie(0);
    cin>>n>>m>>v>>e;
	for(int i=1;i<=n;i++)
	{
		cin>>c[i];
	}
	for(int i=1;i<=n;i++)
	{
		cin>>d[i];
	}
	for(int i=1;i<=n;i++)
	{
		cin>>p[i];
	}
	memset(dis,0x7f,sizeof dis);
	memset(f,0x7f,sizeof f);
	for(int i=1;i<=v;i++)
	dis[i][i]=0;
	for(int i=1;i<=e;i++)
	{
		int a,b;
		double w;
		cin>>a>>b>>w;
		dis[b][a]=dis[a][b]=min(dis[a][b],w);
	}
	for(int k=1;k<=v;k++)
	{
		for(int i=1;i<=v;i++)
		{
			for(int j=1;j<=v;j++)
			{
				if(dis[i][j]>dis[i][k]+dis[k][j])
				dis[i][j]=dis[i][k]+dis[k][j];
			}
		}
	}
	f[1][0][0]=f[1][1][1]=0;
	for(int i=2;i<=n;i++)
	{
		for(int j=0;j<=min(i,m);j++)
		{
			f[i][j][0]=min(f[i-1][j][0]+dis[c[i-1]][c[i]],f[i-1][j][1]+p[i-1]*dis[d[i-1]][c[i]]+(1-p[i-1])*dis[c[i-1]][c[i]]);
			if(j!=0)
			f[i][j][1]=min(f[i-1][j-1][0]+p[i]*dis[c[i-1]][d[i]]+(1-p[i])*dis[c[i-1]][c[i]],f[i-1][j-1][1]+dis[c[i-1]][c[i]]*(1-p[i])*(1-p[i-1])+dis[d[i-1]][c[i]]*(1-p[i-1])*p[i]+dis[c[i-1]][d[i]]*p[i]*(1-p[i-1])+dis[d[i-1]][d[i]]*p[i]*p[i-1]);
		}
	}
	double ans=0x7fffffff;
	for(int i=0;i<=m;i++)
	{
		ans=min({ans,f[n][i][0],f[n][i][1]});
	}
	printf("%.2lf",ans);
 	return 0;
}
2023/8/8 20:37
加载中...