NOIP 2016 换教室 （期望DP）

Description

对于刚上大学的牛牛来说，他面临的第一个问题是如何根据实际情况申请合适的课程。在可以选择的课程中，有 $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）$节课结束后，牛牛就会从这节课的教室出发，选择一条耗费体力最少的路径前往下一节课的教室。现在牛牛想知道，申请哪几门课程可以使他因在教室间移动耗费的体力值的总和的期望值最小，请你帮他求出这个最小值。

Solution

先Floyed预处理出所有点之间的距离，然后考虑期望dp，$dp_{i,j,0/1}$表示选了第$i$节棵，用了$j$次机会，当前是选/不选。具体转移:

f[i][j][0] = min(
f[i-1][j][0]+w[c[i-1]][c[i]],//上一次和这一次都不换课
f[i-1][j][1]+k[i-1]*w[d[i-1]][c[i]]//上一次换课成功
+ (1-k[i-1])*w[c[i-1]][c[i]]);//上一次换课失败

f[i][j][1] = min(
f[i-1][j-1][0]+//这一次换课、上一次不换课
k[i]*w[c[i-1]][d[i]]+//这一次成功
(1-k[i])*w[c[i-1]][c[i]],//这一次失败
f[i-1][j-1][1] + //上一次和这一次都换课
k[i-1]*k[i]*w[d[i-1]][d[i]] +//两次都成功
(1-k[i-1])*k[i]*w[c[i-1]][d[i]] + //上一次失败，这一次成功
k[i-1]*(1-k[i])*w[d[i-1]][c[i]] +//上一次成功，这一次失败 
(1-k[i-1])*(1-k[i])*w[c[i-1]][c[i]]);//两次都失败

Code

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
#define For(i , j , k) for (int i = (int)(j) ; i <= (int)(k) ; ++ i)
#define Fordown(i , j , k) for (int i = (int)(j) ; i >= (int)(k) ; -- i)
#define Set(a , b) memset(a , b , sizeof(a))
#define INF (0x3fffffff)
using namespace std;

double f[2010][2010][2] , k[2010];
int w[2010][2010] , n , m , em , e , v[2010] , c[2010] , d[2010] , vs;

void Floyed()
{
    For(k , 1 , vs)
        For(i , 1 , vs)
            For(j , 1 , vs)
                w[i][j] = min(w[i][k] + w[k][j] , w[i][j]);
}

void Init()
{
    int uu , vv , ww;
    scanf("%d%d%d%d" , &n , &m , &vs , &em);
    For(i , 1 , n)
        scanf("%d" , &c[i]);
    For(i , 1 , n)
        scanf("%d" , &d[i]);
    For(i , 1 , n)
        scanf("%lf" , &k[i]);
    For(i , 1 , vs)
        For(j , 1 , vs)
            w[i][j] = INF;
    For(i , 1 , vs)
        w[i][i] = 0;
    For(i , 1 , em)
    {
        scanf("%d%d%d" , &uu , &vv , &ww);
        if (w[uu][vv] < ww)
            continue;
        w[uu][vv] = w[vv][uu] = ww;
    }
    Floyed();
}

inline double fmin(double a , double b)
{
    return a < b ? a : b;
}

void DP()
{
    For(i , 1 , n + 1)
        For(j , 0 , m + 1)
            f[i][j][0] = f[i][j][1] = INF;
    f[1][0][0] = 0;
    f[1][1][1] = 0;
    For(i , 2 , n)
    {
        f[i][0][0] = f[i - 1][0][0] + w[c[i - 1]][c[i]];
        For(j , 1 , min(m , i))
        {
            f[i][j][0]=fmin(f[i-1][j][0]+w[c[i-1]][c[i]],f[i-1][j][1]+k[i-1]*w[d[i-1]][c[i]]+(1-k[i-1])*w[c[i-1]][c[i]]);
            f[i][j][1]=fmin(f[i-1][j-1][0]+k[i]*w[c[i-1]][d[i]]+(1-k[i])*w[c[i-1]][c[i]],f[i-1][j-1][1]+k[i-1]*k[i]*w[d[i-1]][d[i]]+(1-k[i-1])*k[i]*w[c[i-1]][d[i]]+k[i-1]*(1-k[i])*w[d[i-1]][c[i]]+(1-k[i-1])*(1-k[i])*w[c[i-1]][c[i]]);
        }
    }
}

void Print()
{
    double minn = f[n][0][0];
    For(i , 1 , m)
        minn = fmin(minn , f[n][i][0]),
        minn = fmin(minn , f[n][i][1]);
    printf("%.2lf\n" , minn);
}

int main()
{
#ifndef ONLINE_JUDGE
    freopen("hjs.in" , "r" , stdin);
    freopen("hjs.out" , "w" , stdout);
#endif
    Init();
    DP();
    Print();
    return 0;
}
//今年寒食在商山，山里风光亦可怜：
//稚子就花拈蛱蝶，人家依树系秋千。
//郊原晓绿初经雨，巷陌春阴乍禁烟。
//副使官闲莫惆怅，酒钱犹有撰碑钱。
//--王禹偁《寒食》