P1850 换教室——期望DP

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

在可以选择的课程中，有 $\leq i \leq n1≤i≤n）个时间段上，两节内容相同的课程同时在不同的地点进行，其中，牛牛预先被安排在教室 cic_ici 上课，而另一节课程在教室 did_idi 进行。$

在不提交任何申请的情况下，学生们需要按时间段的顺序依次完成所有的 $did_idi 上课，否则仍然在教室 cic_ici 上课。$

由于更换教室的需求太多，申请不一定能获得通过。通过计算，牛牛发现申请更换第 $kik_iki，并且对于不同课程的申请，被通过的概率是互相独立的。$

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

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

牛牛所在的大学有 $\leq i \leq n-11≤i≤n−1）节课结束后，牛牛就会从这节课的教室出发，选择一条耗费体力最少的路径前往下一节课的教室。$

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

第一行四个整数

第二行 $\leq i \leq n1≤i≤n）个正整数表示 cic_ici，即第 iii 个时间段牛牛被安排上课的教室；保证 1≤ci≤v1 \le c_i \le v1≤ci≤v。$

第三行 $\leq i \leq n1≤i≤n）个正整数表示 did_idi，即第 iii 个时间段另一间上同样课程的教室；保证 1≤di≤v1 \le d_i \le v1≤di≤v。$

第四行 $\leq i \leq n1≤i≤n）个实数表示 kik_iki，即牛牛申请在第 iii 个时间段更换教室获得通过的概率。保证 0≤ki≤10 \le k_i \le 10≤ki≤1。$

接下来 $aj,bj,wja_j, b_j, w_jaj,bj,wj，表示有一条双向道路连接教室 aj,bja_j, b_jaj,bj，通过这条道路需要耗费的体力值是 wjw_jwj；保证 1≤aj,bj≤v1 \le a_j, b_j \le v1≤aj,bj≤v， 1≤wj≤1001 \le w_j \le 1001≤wj≤100。$

保证 $\leq n \leq 20001≤n≤2000，0≤m≤20000 \leq m \leq 20000≤m≤2000，1≤v≤3001 \leq v \leq 3001≤v≤300，0≤e≤900000 \leq e \leq 900000≤e≤90000。$

保证通过学校里的道路，从任何一间教室出发，都能到达其他所有的教室。

保证输入的实数最多包含

输出一行，包含一个实数，四舍五入精确到小数点后恰好

测试数据保证四舍五入后的答案和准确答案的差的绝对值不大于 $\times 10^{-3}4×10−3。（如果你不知道什么是浮点误差，这段话可以理解为：对于大多数的算法，你可以正常地使用浮点数类型而不用对它进行特殊的处理）$

【样例1说明】

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

【提示】

道路中可能会有多条双向道路连接相同的两间教室。也有可能有道路两端连接的是同一间教室。
请注意区分n,m,v,e的意义, n不是教室的数量, m不是道路的数量。

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

特殊性质2:对于所有的 $ki=1k_i= 1ki=1 。$

$dp[i][j][k] 表示第i节课换了j次，上一次换还是没换（k=0，1）最小的期望值；$

$我们根据上节课换没换推出上节课的位置和现在的位置；$

$最短路不用说；$

$因为是期望，所以每次你要乘上成功和不成功的概率；$

$如果没换就直接加；$

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=2010;
const double inf=1e9+10;
int n,m,v,e;
int g[310][310];
int dis[310][310];
int c[maxn],d[maxn];
double k[maxn];
double dp[maxn][maxn][2];
int main()
{
    memset(dis,0x3f,sizeof(dis));
    scanf("%d%d%d%d",&n,&m,&v,&e);
    for(int i=1;i<=n;i++) scanf("%d",&c[i]);
    for(int i=1;i<=n;i++) scanf("%d",&d[i]);
    for(int i=1;i<=n;i++) scanf("%lf",&k[i]);
    
    for(int i=1;i<=e;i++)
    {
        int qw,we,er;
        scanf("%d%d%d",&qw,&we,&er);
        dis[qw][we]=min(er,dis[qw][we]);
        dis[we][qw]=min(er,dis[we][qw]);
    }
    for(int kq=1;kq<=v;kq++) for(int i=1;i<=v;i++) for(int j=1;j<=v;j++)
    {
        if(dis[i][j]>dis[i][kq]+dis[kq][j]) dis[i][j]=dis[i][kq]+dis[kq][j];
    }
    for(int i=0;i<=n;i++) for(int j=0;j<=m;j++) for(int k=0;k<=1;k++) 
    dp[i][j][k]=inf;
    for(int i=1;i<=v;i++) dis[i][i]=dis[i][0]=dis[0][i]=0;
    dp[1][0][0]=0;dp[1][1][1]=0;
    for(int i=2;i<=n;i++)
    {
        dp[i][0][0]=dp[i-1][0][0]+dis[c[i-1]][c[i]];
        for(int j=1;j<=min(i,m);j++)
        {
            double last=min(dp[i-1][j][0]+dis[c[i-1]][c[i]],dp[i-1][j][1]+dis[c[i-1]][c[i]]*(1-k[i-1])+dis[d[i-1]][c[i]]*k[i-1]);
            dp[i][j][0]=min(dp[i][j][0],last);
            double last1=dp[i-1][j-1][0]+dis[c[i-1]][c[i]]*(1-k[i])+dis[c[i-1]][d[i]]*k[i];
            double last2=dp[i-1][j-1][1]+dis[c[i-1]][d[i]]*(1-k[i-1])*k[i]+dis[d[i-1]][d[i]]*k[i]*k[i-1]+dis[c[i-1]][c[i]]*(1-k[i-1])*(1-k[i])+dis[d[i-1]][c[i]]*k[i-1]*(1-k[i]);
            dp[i][j][1]=min(dp[i][j][1],min(last1,last2)); 
        }
    }
    double ans=inf;
    for(int i=0;i<=m;i++) ans=min(ans,min(dp[n][i][0],dp[n][i][1]));
    printf("%.2lf",ans); 
    return 0;
}

double不要用memset,赋值是小数点后面的数

P1850 换教室——期望DP

题目描述

输入格式

输出格式

输入输出样例

说明/提示

猜你喜欢