NOIP2016Day1T3-换教室

概率dp经典题。

3、 换教室

时间限制: 1 Sec 内存限制: 512 MB

题目描述

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

对于刚上大学的牛牛来说，他面临的第一个问题是如何根据实际情况申请合适的课程。在不提交任何申请的情况下，学生们需要按时间段的顺序依次完成所有的n节安排好的课程。如果学生想更换第i节课程的教室，则需要提出申请。若申请通过，学生就可以在第i个时间段去教室di上课，否则仍然在教室di上课。

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

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

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

牛牛所在的大学有v个教室，有e条道路。每条道路连接两间教室，并且是可以双向通行的。

由于道路的长度和拥堵程度不同，通过不同的道路耗费的体力可能会有所不同。当第i（1≤i≤n−1）节课结束后，牛牛就会从这节课的教室出发，选择一条耗费体力最少的路径前往下一节课的教室。现在牛牛想知道，申请哪几门课程可以使他因在教室间移动耗费的体力值的总和的期望值最小，请你帮他求出这个最小值。

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

输入

第一行四个整数n,m,v,e。n表示这个学期内的时间段的数量；m表示牛牛最多可以申请更换多少节课程的教室；v表示牛牛学校里教室的数量；e表示牛牛的学校里道路的数量。

第二行n个正整数，第i（1≤i≤n）个正整数表示ci ，即第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。（如果你不知道什么是浮点误差，这段话可以理解为：对于大多数的算法，你可以正常地使用浮点数类型而不用对它进行特殊的处理）

样例输入

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

样例输出

2.80

提示

【样例1说明】

所有可行的申请方案和期望收益如下表:

【样例 2】

见选手目录下的classroom/classroom2.in与classroom/classroom2.ans。

【提示】

1.道路中可能会有多条双向道路连接相同的两间教室。也有可能有道路两端连接的是同一间教室。

2.请注意区分n,m,v,e的意义，n不是教室的数量，m不是道路的数量。

【子任务】

特殊性质1:图上任意两点ai,bi,ai≠bi间,存在一条耗费体力最少的路径只包含一条道路。

特殊性质2:对于所有的1≤i≤n，ki=1。

这道题比第二题简单。很容易想到Dp，但状态转移方程比较难想到。

用dp[i][j][0..1]表示当前在第i个时间段，申请了j个教室，0表示申请不通过，1表示申请通过，最小耗费体力的期望值。

a[i]表示原来时间段的教室，b[i]表示更换后的教室，c[i]是申请成功的概率。

另外还要用Floyd求出每两个点之间的最短距离记为mat[i][j];

转移：

当前时间段由之前时间段申请和不申请转移。

注意申请要分申请通过和不通过分别计算。

dp[i][j][0]=min(dp[i-1][j][0]+mat[a[i-1]][a[i]],dp[i-1][j][1]+mat[a[i-1]][a[i]](1-c[i-1])+mat[b[i-1]][a[i]]c[i-1]);

dp[i][j][1]=min(dp[i-1][j-1][0]+mat[a[i-1]][a[i]](1-c[i])+mat[a[i-1]][b[i]]c[i],dp[i-1][j-1][1]+mat[a[i-1]][a[i]](1-c[i-1])(1-c[i])+mat[b[i-1]][a[i]]c[i-1](1-c[i])+mat[a[i-1]][b[i]](1-c[i-1])c[i]+mat[b[i-1]][b[i]]c[i-1]c[i]);

Code:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=2010,V=310;
int a[N],b[N];
double d[V][V];
double c[N];
double f[N][N][2];
int cnt=0;
int n,m,v,e;
void init()
{
    for (int i=0;i<V;i++)
        for (int j=0;j<V;j++)
            if (i==j) d[i][j]=0;
            else d[i][j]=1e9;
    for (int i=0;i<N;i++)
        for (int j=0;j<N;j++)
            f[i][j][0]=f[i][j][1]=1e9;
    f[1][0][0]=f[1][1][1]=0;
}
void floyd()
{
    for (int k=1;k<=v;k++)
        for (int i=1;i<=v;i++)
            for (int j=1;j<=v;j++)
                if (d[i][k]+d[k][j]<d[i][j])
                    d[i][j]=d[i][k]+d[k][j];
}
int main()
{
    scanf("%d%d%d%d",&n,&m,&v,&e);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    for (int i=1;i<=n;i++) scanf("%d",&b[i]);
    for (int i=1;i<=n;i++) scanf("%lf",&c[i]);
    init();
    for (int i=1;i<=e;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        if (x==y) continue;
        if (z<d[x][y])
        {
            d[x][y]=z;
            d[y][x]=d[x][y];
        }
    }
    floyd();
    for (int i=2;i<=n;i++)
    {
        f[i][0][0]=f[i-1][0][0]+d[a[i-1]][a[i]];
        for (int j=1;j<=min(m,i);j++)
        {
            f[i][j][0]=min(f[i-1][j][0]+d[a[i-1]][a[i]],f[i-1][j][1]+
					   d[a[i-1]][a[i]]*(1-c[i-1])+d[b[i-1]][a[i]]*c[i-1]);
            f[i][j][1]=f[i-1][j-1][0]+d[a[i-1]][a[i]]*(1-c[i])+d[a[i-1]][b[i]]*c[i];
            f[i][j][1]=min(f[i][j][1],f[i-1][j-1][1]+d[a[i-1]][a[i]]*(1-c[i-1])*(1-c[i])
			           +d[b[i-1]][a[i]]*c[i-1]*(1-c[i])+d[a[i-1]][b[i]]*(1-c[i-1])*c[i]+
					   d[b[i-1]][b[i]]*c[i-1]*c[i]);
        }
    }
    double ans=f[n][0][0];
    for (int i=1;i<=m;i++)
    {
        ans=min(ans,min(f[n][i][0],f[n][i][1]));
    }
    printf("%.2lf",ans);
    return 0;
}