【BZOJ】4720 [Noip2016]换教室 期望DP

题目传送门

这题考的是期望DP，虽然在考的时候我并不会期望DP，但这仍然不能阻挡我骗分的脚步。

至少用floyd和枚举第一段时间是否换课还是可以骗到52分的。

首先是用floyd跑出每两个点之间的最短路，然后就是DP。

我们定义f[i][j][k]表示在第i段时间，申请换课j次，当前是否换课的最小期望。

然后就是对于第i段时间和第i-1段时间是否换课的分类讨论：

1.若第i段时间和第i-1段时间都不申请换课，则f[i][j][0]=min(f[i][j][k],f[i-1][j][0]+map[c[i-1]][c[i]])。

2.若第i段时间不申请换课，第i-1段时间申请换课，则f[i][j][0]=min(f[i][j][0],f[i-1][j][1]+map[d[i-1]][c[i]]*p[i-1]+map[c[i-1]][c[i]]*(1-p[i-1]))。

3.若第i段时间申请换课，第i-1段时间不申请换课，则f[i][j][1]=min(f[i][j][1],f[i-1][j-1][0]+map[c[i-1]][d[i]]*p[i]+map[c[i-1]][c[i]]*(1-p[i]))。

4.若第i段时间和第i-1段时间都申请换课，则f[i][j][1]=min(f[i][j][1],f[i-1][j-1][1]+map[c[i-1]][c[i]]*(1-p[i-1])*(1-p[i])+map[c[i-1]][d[i]]*(1-p[i-1])*p[i]+map[d[i-1]][c[i]]*p[i-1]*(1-p[i])+map[d[i-1]][d[i]]*p[i-1]*p[i])。

最后，在第n段时间内，0到k次换课机会，每次是否换课的所有情况中找一个期望最小的作为答案就行了。

附上AC代码：

#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define N 2010
using namespace std;

int n,m,v,e,c[N],d[N],map[N][N],x,y,w;
double p[N],ans,f[N][N][2],a,b;

void read(int& a){
	static char c=getchar();a=0;int f=1;
	while (!isdigit(c)) {if (c=='-') f=-1;c=getchar();}
	while (isdigit(c)) a=a*10+c-'0',c=getchar();
	a*=f;return;
}

int main(void){
	read(n),read(m),read(v),read(e);
	for (int i=1; i<=n; ++i) read(c[i]);
	for (int i=1; i<=n; ++i) read(d[i]);
	for (int i=1; i<=n; ++i) scanf("%lf",&p[i]);
	memset(map,0x3f,sizeof map);
	for (int i=1; i<=v; ++i) map[i][i]=0;
	for (int i=1; i<=e; ++i){
		read(x),read(y),read(w);
		if (map[x][y]>w) map[x][y]=map[y][x]=w;
	}
	for (int k=1; k<=v; ++k)
		for (int i=1; i<=v; ++i)
			for (int j=1; j<=v; ++j)
				if (i!=j&&i!=k&&j!=k&&map[i][j]>map[i][k]+map[k][j])
					map[i][j]=map[i][k]+map[k][j];
	for (int i=1; i<=n; ++i) for (int j=0; j<=m; ++j) f[i][j][0]=f[i][j][1]=1<<30;
	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){
			a=f[i-1][j][0]+map[c[i-1]][c[i]];
			b=f[i-1][j][1]+map[d[i-1]][c[i]]*p[i-1]+map[c[i-1]][c[i]]*(1-p[i-1]);
			f[i][j][0]=min(f[i][j][0],min(a,b));
			if (j>0){
				a=f[i-1][j-1][0]+map[c[i-1]][d[i]]*p[i]+map[c[i-1]][c[i]]*(1-p[i]);
				b=f[i-1][j-1][1]+map[c[i-1]][c[i]]*(1-p[i-1])*(1-p[i])+
								 map[c[i-1]][d[i]]*(1-p[i-1])*p[i]+
								 map[d[i-1]][c[i]]*p[i-1]*(1-p[i])+
								 map[d[i-1]][d[i]]*p[i-1]*p[i];
				f[i][j][1]=min(f[i][j][1],min(a,b));
			}
		}
	ans=1<<30;
	for (int i=0; i<=m; ++i) ans=min(ans,min(f[n][i][0],f[n][i][1]));
	printf("%.2lf",ans);
	return 0;
}