NOIP2016——换教室（floyd+期望dp）

描述
对于刚上大学的牛牛来说, 他面临的第一个问题是如何根据实际情况中情合适的课程。
在可以选择的课程中,有2n节课程安排在n个时间段上。在第 i ( 1≤ i≤n)个时同段上, 两节内容相同的课程同时在不同的地点进行, 其中, 牛牛预先被安排在教室 ci上课, 而另一节课程在教室 di进行。
在不提交任何申请的情况下,学生们需要按时间段的顺序依次完成所有的n节安排好的课程。如果学生想更换第i节课程的教室,则需要提出中情。若申请通过,学生就可以在第 i个时间段去教室 di上课, 否则仍然在教室 ci上课。
由于更换教室的需求太多, 申请不一定能获得通过。 通过计算, 牛牛发现申请更换第 i节课程的教室时, 中情被通过的概率是一个已知的实数 ki, 并且对于不同课程的申请, 被通过的概率是互相独立的。
学校规定, 所有的申请只能在学期开始前一次性提交, 并且每个人只能选择至多m节课程进行申请。 这意味着牛牛必须一次性决定是否申请更换每节课的教室, 而不能根据某些课程的申请结果来决定其他课程是否申请; 牛牛可以申请白己最希望更换教室的 m门课程,也可以不用完这m个中情的机会,甚至可以一门课程都不申请。
因为不同的课程可能会被安排在不同的教室进行, 所以牛牛需要利用课问时间从一间教室赶到另一间教室。
牛牛所在的大学有 v个教室,有 e条道路。每条道路连接两间教室, 并且是可以双向通行的。 由于道路的长度和拥;i者程度不同, 通过不同的道路耗费的体力可能会有所不同。当第i ( 1≤i≤n-1 )节课结束后,牛牛就会从这节课的教室出发,选择一条耗费体力最少的路径前往下一节课的教室。
现在牛牛想知道,申请哪几门课程可以使他因在教室问移动耗费的体力值的总和的期望值最小,请你帮他求出这个最小值。
输入格式：
第一行四个整数 n,m,v,e 。 n表示这个学期内的时间段的数量; m表示牛牛最多可以申请更换多少节课程的教室; v表示牛牛学校里教室的数量; e表示牛牛的学校里道路的数量。
第二行n个正整数,第 i ( 1≤ i≤ n)个正整数表示c,即第 i个时间段牛牛被安排上课的教室;保证1≤ ci≤ v。
第三行n个正整数,第 i ( 1≤ i≤ n)个正整数表示di,即第 i个时间段另一间上同样课程的教室;保证1≤ di≤ v。
第四行n个实数,第 i ( 1≤ i≤ n)个实数表示ki,即牛牛申请在第 i个时间段更换教室获得通过的概率。保证0≤ ki≤1 。
接下来 e行,每行三个正整数aj,bj,wj,表示有一条双向道路连接教室 aj ,bj ,通过这条道路需要耗费的体力值是 Wj ;保证1≤ aj,bj≤ v, 1≤ wj≤100 。
保证1≤n≤2000, 0≤m≤2000, 1≤v≤300, 0≤ e≤90000。
保证通过学校里的道路,从任何一间教室出发,都能到达其他所有的教室。
保证输入的实数最多包含3位小数。
输出格式：
输出一行,包含一个实数,四舎五入精确到小数点后恰好2位,表示答案。你的
输出必须和标准输出完全一样才算正确。
测试数据保证四舎五入后的答案和准确答案的差的绝对值不大于4 *10^-3 。 (如果你不知道什么是浮点误差, 这段话可以理解为: 对于大多数的算法, 你可以正常地使用浮点数类型而不用对它进行特殊的处理)

我觉得这道题比T2天天爱跑步还要简单很多啊

T2只去打暴力就够了，这道题随便A嘛

因为教室的数量只有300，所以我们直接用floyd求出两两教师之间的距离

如果再多点就只能SPFA了，

首先有n个课，可以申请m次，那么肯定$dp[i][j]$表示前i个申请j次的最优值

考虑到每个课程可以选择申请或者不申请，再加一维0/1表示这一次是否申请（不一定要申请成功）

那么对于第$i$门课程成功的概率就为$p[i]$，不成功的概率就为$(1-p[i])$

首先我们考虑不申请的情况

那么我们考虑第$i-1$节课

有申请和不申请两种情况

不申请则是

$dp[i][j][0]=dp[i-1][j][0]+dis[c[i-1]][c[i]]$ ($c[i],d[i]$的定义如原题)

如果申请了，则还要考虑成功和不成功两种情况的贡献分别乘上各自的概率

为：
$dp[i][j][0]=dp[i-1][j][1]+dis[c[i-1]][c[i]]*(1-p[i-1])+dis[d[i-1]][c[i]]*p[i]$

所以综合起来不申请的情况就是

$dp[i][j][0]=min(dp[i][j][0],min(dp[i-1][j][0]+dis[c[i-1]][c[i]],dp[i-1][j][1]+dis[c[i-1]][c[i]]*(1-p[i-1])+dis[d[i-1]][c[i]]*p[i-1])$

那么我们再来考虑申请的情况：

这里又要分为前面一门课程申请了和没申请两种情况

如果没申请

那么自己就要分为成功和不成功两种：

$dp[i][j][1]=dp[i-1][j-1][0]+dis[c[i-1]][d[i]]*p[i]+dis[c[i-1]][c[i]]*(1-p[i])$

而如果申请了的话就要考虑四种可能情况的贡献：

前一个成功，这一个也成功：

$dis[d[i-1]][d[i]]*p[i-1]*p[i]$

前一个成功，这一个未成功：

$dis[d[i-1]][c[i]]*p[i-1]*(1-p[i]))$

前一个未成功，这一个成功：

$dis[c[i-1]][d[i]]*(1-p[i-1])*p[i]$

前一个未成功，这一个也未成功;

$dis[c[i-1]][c[i]]*(1-p[i-1])*(1-p[i])$

所以申请的情况综合起来就是

$dp[i][j][1]=min(dp[i][j][1],min(dp[i-1][j-1][0]+dis[c[i-1]][d[i]]*p[i]+dis[c[i-1]][c[i]]*(1-p[i]),dp[i-1][j-1][1]+dis[d[i-1]][d[i]]*p[i-1]*p[i]+dis[c[i-1]][d[i]]*(1-p[i-1])*p[i]+dis[c[i-1]][c[i]]*(1-p[i-1])*(1-p[i])+dis[d[i-1]][c[i]]*p[i-1]*(1-p[i])))$

当然这必须要在$j>0$的时候才行

所以说综合起来的递推方程就是

$dp[i][j][0]=min(dp[i][j][0],min(dp[i-1][j][0]+dis[c[i-1]][c[i]],dp[i-1][j][1]+dis[c[i-1]][c[i]]*(1-p[i-1])+dis[d[i-1]][c[i]]*p[i-1]));$

$dp[i][j][1]=min(dp[i][j][1],min(dp[i-1][j-1][0]+dis[c[i-1]][d[i]]*p[i]+dis[c[i-1]][c[i]]*(1-p[i]),dp[i-1][j-1][1]+dis[d[i-1]][d[i]]*p[i-1]*p[i]+dis[c[i-1]][d[i]]*(1-p[i-1])*p[i]+dis[c[i-1]][c[i]]*(1-p[i-1])*(1-p[i])+dis[d[i-1]][c[i]]*p[i-1]*(1-p[i])));$$(if(j>=1))$

这真的有点长啊，但其实理解之后就很简单了

#include<bits/stdc++.h>
using namespace std;
inline int read(){
	char ch=getchar();
	int res=0;
	while(!isdigit(ch)) ch=getchar();
	while(isdigit(ch)) res=(res<<3)+(res<<1)+(ch^48),ch=getchar();
	return res;
}
int n,m,v,e,dis[305][305],c[2005],d[2005];
double p[2005],dp[2005][2005][2];
int main(){
	n=read(),m=read(),v=read(),e=read();
	for(int i=1;i<=n;++i){
		c[i]=read();
	}
	for(int i=1;i<=n;++i){
		d[i]=read();
	}
	for(int i=1;i<=n;++i){
		cin>>p[i];
	}
	memset(dis,127/3,sizeof(dis));
	for(int i=1;i<=e;++i){
		int u=read(),v=read(),w=read();
		dis[u][v]=dis[v][u]=min(w,dis[u][v]);
	}
	for(int k=1;k<=v;++k){
		for(int i=1;i<=v;++i){
			for(int j=1;j<=v;++j){
				if(dis[i][k]+dis[k][j]<dis[i][j]){
					dis[i][j]=dis[i][k]+dis[k][j];
				}
			}
		}
	}
	for(int i=1;i<=v;++i){
		dis[i][i]=0;
	}
	for(int i=1;i<=n;++i){
		for(int j=0;j<=m;++j){
			dp[i][j][0]=dp[i][j][1]=1e30;
		}
	}
	dp[1][0][0]=dp[1][1][1]=0;
	for(register int i=1;i<=n;++i){
		for(register int j=0;j<=m;++j){
			dp[i][j][0]=min(dp[i][j][0],min(dp[i-1][j][0]+dis[c[i-1]][c[i]],dp[i-1][j][1]+dis[c[i-1]][c[i]]*(1-p[i-1])+dis[d[i-1]][c[i]]*p[i-1]));
			if(j>=1)
			dp[i][j][1]=min(dp[i][j][1],min(dp[i-1][j-1][0]+dis[c[i-1]][d[i]]*p[i]+dis[c[i-1]][c[i]]*(1-p[i]),dp[i-1][j-1][1]+dis[d[i-1]][d[i]]*p[i-1]*p[i]+dis[c[i-1]][d[i]]*(1-p[i-1])*p[i]+dis[c[i-1]][c[i]]*(1-p[i-1])*(1-p[i])+dis[d[i-1]][c[i]]*p[i-1]*(1-p[i])));
		}
	}
	double ans=99999999.0;
	for(register int i=0;i<=m;i++)
	{
		ans=min(ans,min(dp[n][i][0],dp[n][i][1]));
	}
	printf("%.2lf",ans);
	return 0;
}