换教室

题目描述

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

在可以选择的课程中,有 2n2n 节课程安排在 nn 个时间段上。在第 ii ( 1 \leq i \leq n1≤i≤n )个时间段上,两节内容相同的课程同时在不同的地点进行,其中,牛牛预先被安排在教室 c_ic
i
​ 上课,而另一节课程在教室 d_id
i
​ 进行。

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

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

学校规定,所有的申请只能在学期开始前一次性提交,并且每个人只能选择至多 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_ic
i
​ ,即第 ii 个时间段牛牛被安排上课的教室;保证 1 \le c_i \le v1≤c
i
​ ≤v 。

第三行 nn 个正整数,第 ii ( 1 \leq i \leq n1≤i≤n )个正整数表示 d_id
i
​ ,即第 ii 个时间段另一间上同样课程的教室;保证 1 \le d_i \le v1≤d
i
​ ≤v 。

第四行 nn 个实数,第 ii ( 1 \leq i \leq n1≤i≤n )个实数表示 k_ik
i
​ ,即牛牛申请在第 ii 个时间段更换教室获得通过的概率。保证 0 \le k_i \le 10≤k
i
​ ≤1 。

接下来 ee 行,每行三个正整数 a_j, b_j, w_ja
j
​ ,b
j
​ ,w
j
​ ,表示有一条双向道路连接教室 a_j, b_ja
j
​ ,b
j
​ ,通过这条道路需要耗费的体力值是 w_jw
j
​ ;保证 1 \le a_j, b_j \le v1≤a
j
​ ,b
j
​ ≤v , 1 \le w_j \le 1001≤w
j
​ ≤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
。 (如果你不知道什么是浮点误差,这段话可以理解为:对于大多数的算法,你可以正常地使用浮点数类型而不用对它进行特殊的处理)

思路

首先这是个经典期望dp

其实这题应该是绿题不该是蓝题

只要知道期望dp的基本套路就能做

期望的计算:如果概率为k的代价为w1,概率为(1-k)的代价为w2,那么期望就是概率乘代价,即

kw1+(1-k)w2

n个时段的期望等于每个时段的期望之和,显然的.

于是我们来看这个题

首先有最短路,floyed暴力预处理x和y之间的距离dis[x][y]

假如到i-1个时段已经走了距离d,那么有可以选择换和不换第i个时段的教室

如果不换,那么显然这一时段的教室只能为ci

这时上一次分为换和不换两种情况,教室有k[i-1]的概率为di和(1-k[i-1])的概率为ci

期望按上面的算就好,教室之间的dis乘上概率

如果换第i个时段的教室,那么有k[i]的概率为di和(1-k[i])的概率为ci

然后上一次的同理

这个时候有k[i]k[i-1]的概率两个教室分别为di和d[i-1]

剩下的情况同理

于是我们设f[i][j][k]为到第i个时段已经换了j个教室的期望,k描述这一次换不换,为0换,为1不换

于是状态转移方程很容易写出

按照上面的描述就好

具体的可以看代码

初值,f[1][1][1]=f[1][0][0]=0,其它为inf,很好理解

答案就是max{f[n][i][k]},i=0…m,k=0…1

代码

#include<cstdio>
#include<iostream>
using namespace std;
const int inf=1<<20;
int n,m,v,e,a[2001],b[2001],dis[301][301];
double ans,f[2001][2],c[2001];
inline char gc()
{
    static char BB[1000001],*S=BB,*T=BB;
    return S==T&&(T=(S=BB)+fread(BB,1,1000000,stdin),S==T)?EOF:*S++;
}
inline int getin()
{
    int x=0;char ch=gc();
    while(ch<'0'||ch>'9')ch=gc();
    while(ch>='0'&&ch<='9')x=x*10+ch-48,ch=gc();
    return x;
}
inline double rgetin()
{
    double x=0;char ch=gc();
    while(ch<'0'||ch>'9')ch=gc();
    while(ch>='0'&&ch<='9')x=x*10+(ch-48),ch=gc();
    if(ch=='.')
    {
        double l=0.1;
        ch=gc();
        while(ch>='0'&&ch<='9')x=x+(ch-48)*l,l/=10,ch=gc();
    }
    return x;
}
int main()
{
    n=getin(),m=getin(),v=getin(),e=getin();
    for(register int i=1;i<=n;i++)a[i]=getin();
    for(register int i=1;i<=n;i++)b[i]=getin();
    for(register int i=1;i<=n;i++)c[i]=rgetin();
    for(register int i=1;i<=v;i++)
        for(register int j=1;j<i;j++)
            dis[i][j]=dis[j][i]=inf;
    for(register int i=1;i<=e;i++)
    {
        register int x,y,w;
        x=getin(),y=getin(),w=getin(),dis[x][y]=dis[y][x]=min(dis[x][y],w);
    }
    for(register int k=1;k<=v;k++)
        for(register int i=1;i<=v;i++)
            for(register int j=1;j<i;j++)
                if(dis[i][k]+dis[k][j]<dis[i][j])
                    dis[i][j]=dis[j][i]=dis[i][k]+dis[k][j];
    for(register int j=0;j<=m;j++)
            f[j][0]=f[j][1]=inf;
    f[0][0]=f[1][1]=0;
    for(register int i=2;i<=n;i++)
    {
        register int mn=min(i,m);
        register double w1=dis[a[i-1]][a[i]],w2=dis[b[i-1]][a[i]]*c[i-1]+dis[a[i-1]][a[i]]*(1-c[i-1]),w3=dis[a[i-1]][b[i]]*c[i]+dis[a[i-1]][a[i]]*(1-c[i]),w4=dis[a[i-1]][a[i]]*(1-c[i-1])*(1-c[i])+dis[a[i-1]][b[i]]*(1-c[i-1])*c[i]+dis[b[i-1]][a[i]]*(1-c[i])*c[i-1]+dis[b[i-1]][b[i]]*c[i-1]*c[i];
        for(register int j=mn;j>=0;j--)
        {
            f[j][0]=min(f[j][0]+w1,f[j][1]+w2);
            if(j)f[j][1]=min(f[j-1][0]+w3,f[j-1][1]+w4);
        }
    }
    ans=inf;
    for(register int i=0;i<=m;i++)
        ans=min(ans,min(f[i][0],f[i][1]));
    printf("%.2lf",ans);
}

猜你喜欢

转载自blog.csdn.net/eric1561759334/article/details/80704681