BZOJ4720 [Noip2016]换教室

f[i][j][k]表示前i门课，申请了j次，第i门申请或者不申请

转移即可

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<map>
#include<bitset>
#include<queue>
#include<set>
#include<stack>
using namespace std;
#define MAXN 2010
#define MAXM 310
#define ll long long
#define eps 1e-8
#define INF 1000000000
#define MOD 1000000007
#define min(a,b) (a)<(b)?(a):(b)
int dis[MAXM][MAXM];
double p[MAXN];
double f[MAXN][MAXN][2];
int c[MAXN][2];
int n,m,v,e;
int main(){
	int i,j,k,x,y,z;
	scanf("%d%d%d%d",&n,&m,&v,&e);
	memset(dis,0x3f,sizeof(dis));
	for(i=1;i<=v;i++){
		dis[i][i]=0;
	}
	
	for(i=1;i<=n;i++){
		scanf("%d",&c[i][0]);
	}
	for(i=1;i<=n;i++){
		scanf("%d",&c[i][1]);
	}
	for(i=1;i<=n;i++){
		scanf("%lf",&p[i]);
	}
	for(i=1;i<=e;i++){
		scanf("%d%d%d",&x,&y,&z);
		dis[x][y]=min(dis[x][y],z);
		dis[y][x]=min(dis[y][x],z);
	}
	
	for(k=1;k<=v;k++){
		for(i=1;i<=v;i++){
			for(j=1;j<=v;j++){
				if(i!=k&&i!=j&&k!=j){
					dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]);
				}
			}
		}
	}
	for(i=1;i<=n;i++){
		for(j=0;j<=m;j++){
			f[i][j][0]=f[i][j][1]=1e60;
		}
	}
	f[1][0][0]=0;
	f[1][1][1]=0;
	for(i=2;i<=n;i++){
		f[i][0][0]=f[i-1][0][0]+dis[c[i-1][0]][c[i][0]];
		f[i][0][1]=1e60;
		double t1=p[i-1]*dis[c[i-1][1]][c[i][0]]+(1-p[i-1])*dis[c[i-1][0]][c[i][0]];
		double t2=p[i]*dis[c[i-1][0]][c[i][1]]+(1-p[i])*dis[c[i-1][0]][c[i][0]];
		double t3=p[i-1]*dis[c[i-1][1]][c[i][1]]+(1-p[i-1])*dis[c[i-1][0]][c[i][1]];
		for(j=1;j<=m&&j<=i;j++){
			//*
			f[i][j][0]=min(f[i-1][j][0]+dis[c[i-1][0]][c[i][0]],f[i-1][j][1]+t1);
			f[i][j][1]=min(f[i-1][j-1][0]+t2,f[i-1][j-1][1]+p[i]*(t3)+(1-p[i])*(t1));
			//*/
		}
	}
	double ans=1e60;
	for(i=0;i<=m;i++){
		ans=min(ans,min(f[n][i][0],f[n][i][1]));
	}
	printf("%.2lf\n",ans);
	return 0;
}

/*
3 2 3 3
2 1 2
1 2 1
0.8 0.2 0.5
1 2 5
1 3 3
2 3 1

*/