P1850 [NOIP2016 提高组] 换教室

P1850 [NOIP2016 提高组] 换教室

题目链接

思路

先 Floyd，然后期望 dp ,$f[i][j][0/1]$ 表示第 $i$ 节课已经换了 $j$ 次教室，这一次换不换。

代码

#include<algorithm>
#include<cstdio>
#include<cstring>
int n,m,e,v,a[2005],b[2005],c,d,w,f[305][305];double k[2005],dp[2][2005][2],ans=200000000;
int main(){
	memset(f,0x3f,sizeof(f));
	scanf("%d%d%d%d",&n,&m,&v,&e);
	for(int i=0;i<=m;i++)dp[0][i][0]=dp[1][i][0]=dp[0][i][1]=dp[1][i][1]=200000000.0;
	dp[1][0][0]=dp[1][1][1]=0;
	for(int i=1;i<=v;i++)f[i][i]=0;
	for(int i=1;i<=n;i++)scanf("%d",&a[i]);
	for(int i=1;i<=n;i++)scanf("%d",&b[i]);
	for(int i=1;i<=n;i++)scanf("%lf",&k[i]);
	while(e--)scanf("%d%d%d",&c,&d,&w),f[c][d]=f[d][c]=std::min(f[c][d],w);
	for(int kk=1;kk<=v;kk++)
		for(int i=1;i<=v;i++)
			for(int j=1;j<=v;j++)f[i][j]=std::min(f[i][j],f[i][kk]+f[kk][j]);//Floyd
	for(int i=2;i<=n;i++)
		for(int j=0;j<=std::min(i,m);j++){
			dp[i&1][j][0]=dp[i&1^1][j][0]+f[a[i-1]][a[i]];
			if(j>0)
				dp[i&1][j][0]=std::min(dp[i&1][j][0],dp[i&1^1][j][1]+k[i-1]*f[b[i-1]][a[i]]+(1-k[i-1])*f[a[i-1]][a[i]]),
				dp[i&1][j][1]=dp[i&1^1][j-1][0]+k[i]*f[a[i-1]][b[i]]+(1-k[i])*f[a[i-1]][a[i]];
			if(j>1)
				dp[i&1][j][1]=std::min(dp[i&1][j][1],dp[i&1^1][j-1][1]+k[i]*k[i-1]*f[b[i-1]][b[i]]+(1-k[i])*k[i-1]*f[b[i-1]][a[i]]+k[i]*(1-k[i-1])*f[a[i-1]][b[i]]+(1-k[i])*(1-k[i-1])*f[a[i-1]][a[i]]);
		}
	for(int i=0;i<=m;i++)ans=std::min(ans,std::min(dp[n&1][i][0],dp[n&1][i][1]));
	return printf("%.2f",ans),0;
}