[期望dp][floyd] 洛谷 P1850
题目描述
对于刚上大学的牛牛来说,他面临的第一个问题是如何根据实际情况申请合适的课程。
在可以选择的课程中,有 2n2n 节课程安排在 nn 个时间段上。在第 ii(1 \leq i \leq n1≤i≤n)个时间段上,两节内容相同的课程同时在不同的地点进行,其中,牛牛预先被安排在教室 c_ici 上课,而另一节课程在教室 d_idi 进行。
在不提交任何申请的情况下,学生们需要按时间段的顺序依次完成所有的 nn 节安排好的课程。如果学生想更换第 ii节课程的教室,则需要提出申请。若申请通过,学生就可以在第 ii 个时间段去教室 d_idi 上课,否则仍然在教室 c_ici 上课。
由于更换教室的需求太多,申请不一定能获得通过。通过计算,牛牛发现申请更换第 ii 节课程的教室时,申请被通过的概率是一个已知的实数 k_iki,并且对于不同课程的申请,被通过的概率是互相独立的。
学校规定,所有的申请只能在学期开始前一次性提交,并且每个人只能选择至多 mm 节课程进行申请。这意味着牛牛必须一次性决定是否申请更换每节课的教室,而不能根据某些课程的申请结果来决定其他课程是否申请;牛牛可以申请自己最希望更换教室的 mm 门课程,也可以不用完这 mm 个申请的机会,甚至可以一门课程都不申请。
因为不同的课程可能会被安排在不同的教室进行,所以牛牛需要利用课间时间从一间教室赶到另一间教室。
牛牛所在的大学有 vv 个教室,有 ee 条道路。每条道路连接两间教室,并且是可以双向通行的。由于道路的长度和拥堵程度不同,通过不同的道路耗费的体力可能会有所不同。 当第 ii(1 \leq i \leq n-11≤i≤n−1)节课结束后,牛牛就会从这节课的教室出发,选择一条耗费体力最少的路径前往下一节课的教室。
现在牛牛想知道,申请哪几门课程可以使他因在教室间移动耗费的体力值的总和的期望值最小,请你帮他求出这个最小值。
输入输出格式
输入格式:
第一行四个整数 n,m,v,en,m,v,e。nn 表示这个学期内的时间段的数量;mm 表示牛牛最多可以申请更换多少节课程的教室;vv 表示牛牛学校里教室的数量;ee表示牛牛的学校里道路的数量。
第二行 nn 个正整数,第 ii(1 \leq i \leq n1≤i≤n)个正整数表示 c_ici,即第 ii 个时间段牛牛被安排上课的教室;保证 1 \le c_i \le v1≤ci≤v。
第三行 nn 个正整数,第 ii(1 \leq i \leq n1≤i≤n)个正整数表示 d_idi,即第 ii 个时间段另一间上同样课程的教室;保证 1 \le d_i \le v1≤di≤v。
第四行 nn 个实数,第 ii(1 \leq i \leq n1≤i≤n)个实数表示 k_iki,即牛牛申请在第 ii 个时间段更换教室获得通过的概率。保证 0 \le k_i \le 10≤ki≤1。
接下来 ee 行,每行三个正整数 a_j, b_j, w_jaj,bj,wj,表示有一条双向道路连接教室 a_j, b_jaj,bj,通过这条道路需要耗费的体力值是 w_jwj;保证 1 \le a_j, b_j \le v1≤aj,bj≤v, 1 \le w_j \le 1001≤wj≤100。
保证 1 \leq n \leq 20001≤n≤2000,0 \leq m \leq 20000≤m≤2000,1 \leq v \leq 3001≤v≤300,0 \leq e \leq 900000≤e≤90000。
保证通过学校里的道路,从任何一间教室出发,都能到达其他所有的教室。
保证输入的实数最多包含 33 位小数。
输出格式:
输出一行,包含一个实数,四舍五入精确到小数点后恰好22位,表示答案。你的输出必须和标准输出完全一样才算正确。
测试数据保证四舍五入后的答案和准确答案的差的绝对值不大于 4 \times 10^{-3}4×10−3。 (如果你不知道什么是浮点误差,这段话可以理解为:对于大多数的算法,你可以正常地使用浮点数类型而不用对它进行特殊的处理)
输入输出样例
说明
【样例1说明】
所有可行的申请方案和期望收益如下表:
【提示】
- 道路中可能会有多条双向道路连接相同的两间教室。 也有可能有道路两端连接的是同一间教室。
-
请注意区分n,m,v,e的意义, n不是教室的数量, m不是道路的数量。
特殊性质1:图上任意两点 a_iai, b_ibi, a_iai≠ b_ibi间,存在一条耗费体力最少的路径只包含一条道路。
特殊性质2:对于所有的 1≤ i≤ n1≤i≤n, k_i= 1ki=1 。
题解
- 一开始显然设dp[i][j]为走完前i个教室, 换了j次的最优方案
- 然后发现很难去转移,因为不知道上一个教室的位置,也就是有没有申请成功
- 所以可以改为dp[i][j][k]为走完前i个教室,换了j次,第i个教室我们有没有选择换教室的最优方案
- 首先,考虑不换的情况,也就是k=0
- dp[i][j][0]=min(dp[i][j][0],min(dp[i-1][j][0]+f[c[i-1]][c[i]],dp[i-1][j][1]+f[c[i-1]][c[i]]*(1-k[i-1])+f[d[i-1]][c[i]]*k[i-1]))
- 那么这样就有两种情况了,①是i-1没有换教室,那么i是确定的,那么只有一种可能就是f[c[i-1]][c[i]]
- ②是i-1换了教室,那么i是确定的,那么有两种情况就是换成功了,和没换成功,代价就是距离长度乘上概率
- 然后再考虑换的情况,也就是k=1
-
dp[i][j][1]=min(dp[i][j][1],min(dp[i-1][j-1][0]+f[c[i-1]][c[i]]*(1-k[i])+f[c[i-1]][d[i]]*k[i],dp[i-1][j-1][1]+f[d[i-1]][d[i]]*k[i-1]*k[i]+f[d[i-1]][c[i]]*k[i-1]*(1-k[i])+f[c[i-1]][d[i]]*k[i]*(1-k[i-1])+f[c[i-1]][c[i]]*(1-k[i-1])*(1-k[i])))
- 其实就是跟上面类似
- ①i-1没换
- ⑴ i换成功了 ⑵ i没换成功
- ②i-1也换
- ⑴ i-1成功 i成功 ⑵ i-1成功 i不成功 ⑶ i-1不成功 i成功 ⑷ i-1不成功 i不成功
- 综上所述,其实就是一道恶心的期望dp
- 还要记住,要用Floyd预处理最短距离(最短路)
代码
1 #include <cstdio> 2 #include <iostream> 3 #include <cstring> 4 using namespace std; 5 const double inf=1e17+10; 6 int n,m,v,e,c[2010],d[2010],f[310][310]; 7 double k[2010],dp[2010][2010][2],ans; 8 int main() 9 { 10 memset(f,63,sizeof(f)); 11 scanf("%d%d%d%d",&n,&m,&v,&e); 12 for (int i=1;i<=n;i++) scanf("%d",&c[i]); 13 for (int i=1;i<=n;i++) scanf("%d",&d[i]); 14 for (int i=1;i<=n;i++) scanf("%lf",&k[i]); 15 for (int i=1,x,y,d;i<=e;i++) scanf("%d%d%d",&x,&y,&d),f[x][y]=f[y][x]=min(f[x][y],d); 16 for (int a=1;a<=v;a++) 17 for (int i=1;i<=v;i++) 18 for (int j=1;j<=v;j++) 19 if (i!=j&&i!=a&&j!=a) f[i][j]=min(f[i][j],f[i][a]+f[a][j]); 20 for (int i=1;i<=v;i++) f[i][i]=0; 21 for (int i=0;i<=n;i++) for (int j=0;j<=m;j++) dp[i][j][0]=dp[i][j][1]=inf; 22 dp[1][0][0]=dp[1][1][1]=0; 23 for (int i=2;i<=n;i++) 24 { 25 dp[i][0][0]=dp[i-1][0][0]+f[c[i-1]][c[i]]; 26 for (int j=1;j<=min(i,m);j++) 27 { 28 dp[i][j][0]=min(dp[i][j][0],min(dp[i-1][j][0]+f[c[i-1]][c[i]],dp[i-1][j][1]+f[c[i-1]][c[i]]*(1-k[i-1])+f[d[i-1]][c[i]]*k[i-1])); 29 dp[i][j][1]=min(dp[i][j][1],min(dp[i-1][j-1][0]+f[c[i-1]][c[i]]*(1-k[i])+f[c[i-1]][d[i]]*k[i], 30 dp[i-1][j-1][1]+f[d[i-1]][d[i]]*k[i-1]*k[i]+f[d[i-1]][c[i]]*k[i-1]*(1-k[i])+f[c[i-1]][d[i]]*k[i]*(1-k[i-1])+f[c[i-1]][c[i]]*(1-k[i-1])*(1-k[i]))); 31 } 32 } 33 ans=inf; 34 for (int i=0;i<=m;i++) ans=min(ans,min(dp[n][i][0],dp[n][i][1])); 35 printf("%.2lf",ans); 36 }