【洛谷P1850】换教室

题目

对于刚上大学的牛牛来说，他面临的第一个问题是如何根据实际情况申请合适的课程。

在可以选择的课程中，有 \(2n\) 节课程安排在 \(n\) 个时间段上。在第 \(i\)（\(1 \leq i \leq n\)）个时间段上，两节内容相同的课程同时在不同的地点进行，其中，牛牛预先被安排在教室 \(c_i\) 上课，而另一节课程在教室 \(d_i\) 进行。

在不提交任何申请的情况下，学生们需要按时间段的顺序依次完成所有的 \(n\) 节安排好的课程。如果学生想更换第 \(i\) 节课程的教室，则需要提出申请。若申请通过，学生就可以在第 \(i\) 个时间段去教室 \(d_i\) 上课，否则仍然在教室 \(c_i\) 上课。

由于更换教室的需求太多，申请不一定能获得通过。通过计算，牛牛发现申请更换第 \(i\) 节课程的教室时，申请被通过的概率是一个已知的实数 \(k_i\)，并且对于不同课程的申请，被通过的概率是互相独立的。

学校规定，所有的申请只能在学期开始前一次性提交，并且每个人只能选择至多 \(m\) 节课程进行申请。这意味着牛牛必须一次性决定是否申请更换每节课的教室，而不能根据某些课程的申请结果来决定其他课程是否申请；牛牛可以申请自己最希望更换教室的 \(m\) 门课程，也可以不用完这 \(m\) 个申请的机会，甚至可以一门课程都不申请。

因为不同的课程可能会被安排在不同的教室进行，所以牛牛需要利用课间时间从一间教室赶到另一间教室。

牛牛所在的大学有 \(v\) 个教室，有 \(e\) 条道路。每条道路连接两间教室，并且是可以双向通行的。由于道路的长度和拥堵程度不同，通过不同的道路耗费的体力可能会有所不同。 当第 \(i\)（\(1 \leq i \leq n-1\)）节课结束后，牛牛就会从这节课的教室出发，选择一条耗费体力最少的路径前往下一节课的教室。

现在牛牛想知道，申请哪几门课程可以使他因在教室间移动耗费的体力值的总和的期望值最小，请你帮他求出这个最小值。

思路

发现点数不超过 \(300\)，所以先 Floyd 求出全源最短路径。
设 \(f[i][j][0/1]\) 表示前 \(i\) 节课，申请了 \(j\) 次，其中第 \(i\) 次是否申请的最小期望体力。
那么分类讨论第 \(i-1\) 节课是否申请，转移时加上相应距离期望即可。
时间复杂度 \(O(nm+v^3)\)。

代码

#include <bits/stdc++.h>
using namespace std;

const int N=310,M=2010;
int n,m,WYCtql,XJQtql,dis[N][N],a[M],b[M];
double ans,p[M],f[M][M][2];

int main()
{
	memset(dis,0x3f3f3f3f,sizeof(dis));
	scanf("%d%d%d%d",&WYCtql,&XJQtql,&n,&m);
	for (int i=1;i<=WYCtql;i++) scanf("%d",&a[i]);
	for (int i=1;i<=WYCtql;i++) scanf("%d",&b[i]);
	for (int i=1;i<=WYCtql;i++) scanf("%lf",&p[i]);
	for (int i=1,x,y,z;i<=m;i++)
	{
		scanf("%d%d%d",&x,&y,&z);
		dis[x][y]=dis[y][x]=min(dis[y][x],z);
	}
	for (int i=1;i<=n;i++)
		dis[i][i]=0;
	for (int k=1;k<=n;k++)
		for (int i=1;i<=n;i++)
			for (int j=1;j<=n;j++)
				if (dis[i][j]>dis[i][k]+dis[k][j])
					dis[i][j]=dis[i][k]+dis[k][j];
	n=WYCtql; m=XJQtql;
	for (int i=0;i<=n;i++)
		for (int j=0;j<=m+1;j++)
			f[i][j][1]=f[i][j][0]=1000000000000.0;
	f[1][1][0]=f[1][2][1]=0;
	for (int i=2;i<=n;i++)
		for (int j=1;j<=m+1;j++)
		{
			double d1=1.0*dis[a[i-1]][a[i]],d2=1.0*dis[a[i-1]][b[i]],d3=1.0*dis[b[i-1]][a[i]],d4=1.0*dis[b[i-1]][b[i]];
			f[i][j][0]=min(f[i-1][j][0]+d1,f[i-1][j][1]+d1*(1-p[i-1])+d3*p[i-1]);
			f[i][j][1]=min(f[i-1][j-1][0]+d1*(1-p[i])+d2*p[i],f[i-1][j-1][1]+d1*(1-p[i-1])*(1-p[i])+d2*(1-p[i-1])*p[i]+d3*p[i-1]*(1-p[i])+d4*p[i-1]*p[i]);
		}
	ans=1000000000000.0;
	for (int i=1;i<=m+1;i++)
		ans=min(ans,min(f[n][i][0],f[n][i][1]));
	printf("%.2lf",ans);
	return 0;
}