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

题目描述

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

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

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

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

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

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

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

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

输入输出样例

输入样例#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求出的距离，列出转移方程：

　　

　　

#include<cmath>
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define yy p[i]*p[i-1]
#define yn p[i-1]*(1.0-p[i])
#define ny (1.0-p[i-1])*p[i]
#define nn (1.0-p[i-1])*(1.0-p[i])
using namespace std;

const int maxn=2009;
double f[maxn][maxn][3];
double p[maxn],ans=1.0*1e9;
int n,m,v,e;
int c[maxn],d[maxn],dis[maxn][maxn];
void floyd()
{
    for(int k=1; k<=v; k++)
    {
        for(int i=1; i<=v; i++)
        {
            for(int j=1; j<=v; j++)
            {
                if(dis[i][j]>dis[i][k]+dis[k][j])
                {
                    dis[i][j]=dis[i][k]+dis[k][j];
                }
            }
        }
    }
}
int main()
{
    cin>>n>>m>>v>>e;
    for(int i=1; i<=n; i++)
        cin>>c[i];
    for(int i=1; i<=n; i++)
        cin>>d[i];
    for(int i=1; i<=n; i++)
        cin>>p[i];
    for(int i=1; i<=v+10; i++)
        for(int j=0; j<=v+10; j++)
            dis[i][j]=1e9;
    for(int i=0; i<=v; i++)
        dis[i][i]=0;
    for(int i=1; i<=e; i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        dis[a][b]=dis[b][a]=min(dis[a][b],c);
    }
    floyd();
    for(int i=1; i<=n; i++)
    {
        for(int j=0; j<=m; j++)
        {
            f[i][j][0]=f[i][j][1]=1.0*1e9;
        }
    }
    f[1][0][0]=f[1][1][1]=0;
    for(int i=2; i<=n; i++)
    {
        for(int j=0; j<=min(m,i); j++)
        {
            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] );
            if(j>0)
            {
                f[i][j][1]=min( f[i-1][j-1][0] + dis[c[i-1]][d[i]]*p[i]
                                + dis[c[i-1]][c[i]]*(1.0-p[i]),

                                f[i-1][j-1][1] + dis[d[i-1]][c[i]]*yn
                                + dis[c[i-1]][c[i]]*nn
                                + dis[d[i-1]][d[i]]*yy
                                + dis[c[i-1]][d[i]]*ny    );
            }
        }
    }
    for(int i=0; i<=m; i++)
    {
        ans=min(ans,min(f[n][i][0],f[n][i][1]));
    }
    printf("%.2lf",ans);
    return 0;
}
/*
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
*/