NOIp2016Day1T3--换教室--概率dp+floyd

最新推荐文章于 2022-01-14 21:50:32 发布

dcvook6151

最新推荐文章于 2022-01-14 21:50:32 发布

阅读量119

点赞数

原文链接：http://www.cnblogs.com/Beckinsale/p/7532851.html

版权

题目描述

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

在可以选择的课程中，有

在不提交任何申请的情况下，学生们需要按时间段的顺序依次完成所有的

由于更换教室的需求太多，申请不一定能获得通过。通过计算，牛牛发现申请更换第

学校规定，所有的申请只能在学期开始前一次性提交，并且每个人只能选择至多

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

牛牛所在的大学有

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

输入输出样例

输入样例#1：

　　3 2 3 3

　　2 1 2

　　1 2 1

　　0.8 0.2 0.5

　　1 2 5

　　1 3 3

　　2 3 1

输出样例#1：

　　2.80

题解：

　　首先，考虑对于所有的教室与道路，地图本身是存在最短路的，我们先用floyd求出任意两点之间的最短距离。

　　其次，针对所有事件可能发生的情况：

　　　　1.当前的课不换的情况：

　　　　　　(1)上一节课也没换；

　　　　　　(2)上一节课换了----成功or不成功；

　　　　2.当前的课换的情况：

　　　　　　(1)当前课成功换了：

　　　　　　　　a.上一节课换了----上一节课成功or不成功

　　　　　　　　b.上一节课没换；

　　　　　　(2)当前的课换了失败：

　　　　　　　　a.上一节课换了----上一节课成功or不成功

　　　　　　　　b.上一节课没换；

　　最后，按步就班地结合题中给出的概率，以及我们用floyd求出的距离，列出转移方程：

 1 #include<cmath>
 2 #include<iostream>
 3 #include<cstdio>
 4 #include<algorithm>
 5 #include<cstring>
 6 #define yy p[i]*p[i-1]
 7 #define yn p[i-1]*(1.0-p[i])
 8 #define ny (1.0-p[i-1])*p[i]
 9 #define nn (1.0-p[i-1])*(1.0-p[i])
10 using namespace std;
11 
12 const int maxn=2009;
13 double f[maxn][maxn][3];
14 double p[maxn],ans=1.0*1e9;
15 int n,m,v,e;
16 int c[maxn],d[maxn],dis[maxn][maxn];
17 void floyd()
18 {
19     for(int k=1; k<=v; k++)
20     {
21         for(int i=1; i<=v; i++)
22         {
23             for(int j=1; j<=v; j++)
24             {
25                 if(dis[i][j]>dis[i][k]+dis[k][j])
26                 {
27                     dis[i][j]=dis[i][k]+dis[k][j];
28                 }
29             }
30         }
31     }
32 }
33 int main()
34 {
35     cin>>n>>m>>v>>e;
36     for(int i=1; i<=n; i++)
37         cin>>c[i];
38     for(int i=1; i<=n; i++)
39         cin>>d[i];
40     for(int i=1; i<=n; i++)
41         cin>>p[i];
42     for(int i=1; i<=v+10; i++)
43         for(int j=0; j<=v+10; j++)
44             dis[i][j]=1e9;
45     for(int i=0; i<=v; i++)
46         dis[i][i]=0;
47     for(int i=1; i<=e; i++)
48     {
49         int a,b,c;
50         cin>>a>>b>>c;
51         dis[a][b]=dis[b][a]=min(dis[a][b],c);
52     }
53     floyd();
54     for(int i=1; i<=n; i++)
55     {
56         for(int j=0; j<=m; j++)
57         {
58             f[i][j][0]=f[i][j][1]=1.0*1e9;
59         }
60     }
61     f[1][0][0]=f[1][1][1]=0;
62     for(int i=2; i<=n; i++)
63     {
64         for(int j=0; j<=min(m,i); j++)
65         {
66             f[i][j][0]=min( f[i-1][j][0] + dis[c[i-1]][c[i]] , f[i-1][j][1] + dis[c[i-1]][c[i]]*(1.0-p[i-1]) + dis[d[i-1]][c[i]]*p[i-1] );
67             if(j>0)
68             {
69                 f[i][j][1]=min( f[i-1][j-1][0] + dis[c[i-1]][d[i]]*p[i]
70                                 + dis[c[i-1]][c[i]]*(1.0-p[i]),
71 
72                                 f[i-1][j-1][1] + dis[d[i-1]][c[i]]*yn
73                                 + dis[c[i-1]][c[i]]*nn
74                                 + dis[d[i-1]][d[i]]*yy
75                                 + dis[c[i-1]][d[i]]*ny    );
76             }
77         }
78     }
79     for(int i=0; i<=m; i++)
80     {
81         ans=min(ans,min(f[n][i][0],f[n][i][1]));
82     }
83     printf("%.2lf",ans);
84     return 0;
85 }
86 /*
87 3 2 3 3
88 2 1 2
89 1 2 1
90 0.8 0.2 0.5
91 1 2 5
92 1 3 3
93 2 3 1
94 */