洛谷 P1850 换教室

题意：对于每一个时刻都有一个成功换教室的概率p，两个教室之间有不同的距离，问最小要走的路程的期望是多少。

之前常常有疑惑，为什么期望还有最大和最小？期望不是一个平均值吗？

原因是期望dp中的概率p在从中作祟，也就是可以理解为：通常一个dp某一步产生的贡献可能是一个c[i]，而现在，某一步产生的贡献就是p[i]*c[i]了。

所以对于不同的选择得到的期望，当然是有大小的咯。

就好比，一个学霸和一个学渣在做10道选择题，每题4个选项：学霸每题都能排除两个错误选项，而学渣每题都排除不了一个错误选项。请问他们的期望得分一样吗？当然是不同的。

而他们之前的人生就如同动态规划，学霸的dp取的是max，学渣的dp取的是min，所以，最后的期望值当然有最大和最小之分。

对于这题来说，预处理一个最短路长度后，顺着dp即可。

#include <bits/stdc++.h>
using namespace std;
int n,m,v,e,x,y;
int c[2005],d[2005]; 
double p[2005],f[2005][2005][2],a[305][305];
double z,inf,ans;
//f[i][j][0/1]:前i节课，换了j次教室，第j次有没有换教室 
int main(){
	inf=2e9;
	scanf("%d%d%d%d",&n,&m,&v,&e);
	for (register int i=1; i<=n; ++i) scanf("%d",&c[i]);
	for (register int i=1; i<=n; ++i) scanf("%d",&d[i]);
	for (register int i=1; i<=n; ++i) scanf("%lf",&p[i]);
	
	for (register int i=1; i<=v; ++i)
	for (register int j=1; j<=v; ++j) a[i][j]=inf;
	for (register int i=1; i<=v; ++i) a[i][i]=0;
	for (register int i=1; i<=e; ++i)
	{
		scanf("%d%d%lf",&x,&y,&z);
		a[x][y]=a[y][x]=min(a[x][y],z);	
	}
	for (register int k=1; k<=v; ++k)
	for (register int i=1; i<=v; ++i)
	for (register int j=1; j<=v; ++j) if (i!=j && i!=k && j!=k) a[i][j]=min(a[i][j],a[i][k]+a[k][j]);
	
	for (register int i=1; i<=n; ++i)
	for (register int j=0; j<=m; ++j) f[i][j][0]=f[i][j][1]=inf;
	f[1][0][0]=f[1][1][1]=0;
	for (register int i=2; i<=n; ++i)
	for (register int j=0; j<=min(i,m); ++j)
	{
		f[i][j][0]=min(f[i-1][j][0]+a[c[i-1]][c[i]],f[i-1][j][1]+p[i-1]*a[d[i-1]][c[i]]+(1-p[i-1])*a[c[i-1]][c[i]]);
		if (j) f[i][j][1]=min(f[i-1][j-1][0]+p[i]*a[c[i-1]][d[i]]+(1-p[i])*a[c[i-1]][c[i]],
		f[i-1][j-1][1]+ p[i-1]*p[i]*a[d[i-1]][d[i]]+p[i-1]*(1-p[i])*a[d[i-1]][c[i]]+ 
		(1-p[i-1])*p[i]*a[c[i-1]][d[i]]+(1-p[i-1])*(1-p[i])*a[c[i-1]][c[i]]);
	}
	ans=inf;
	for (register int i=0; i<=m; ++i) ans=min(ans,min(f[n][i][0],f[n][i][1]));
	printf("%.2f\n",ans);
return 0;
}