洛谷——P1850 换教室

P1850 换教室

有 2n 节课程安排在 nn 个时间段上。在第 i个时间段上，两节内容相同的课程同时在不同的地点进行，其中，牛牛预先被安排在教室 $c_i$​ 上课，而另一节课程在教室 $d_i$ 进行。

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

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

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

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

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

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

设$dp[i][j][0]$表示前i节课换了j次且这一次没换的耗费体力值得期望和的最小值，

同理$dp[i][j][1]$表示前i节课换了j次且这一次换了的耗费体力值得期望和的最小值；

首先你换教室有可能成功，当然也有可能失败，那么你这一次的概率就等于上一次换或不换成功不成功，这一次换或不换成功不成功期望路径和的最小值。

分类讨论即可。

#include<bits/stdc++.h>

#define N 2005
using namespace std;

int n,m,V,E,c[N],d[N],dis[N][N];
double k[N],ans,dp[][][];

int main()
{
    scanf("%d%d%d%d",&n,&m,&V,&E);
    for(int i=;i<=n;i++) scanf("%d",&c[i]);
    for(int i=;i<=n;i++) scanf("%d",&d[i]);
    for(int i=;i<=n;i++) scanf("%lf",&k[i]);

    memset(dis,0x3f,sizeof(dis));
    for(int u,v,w,i=;i<=E;i++){
        scanf("%d%d%d",&u,&v,&w);
        dis[u][v]=dis[v][u]=min(w,dis[u][v]);
    }
    //Floyd预处理
    for(int K=;K<=V;K++){
        for(int i=;i<=V;i++){
            for(int j=;j<=V;j++){
                dis[i][j]=min(dis[i][K]+dis[K][j],dis[i][j]);
            }
        }
    }

    for(int i=;i<=V;i++) dis[i][i]=dis[i][]=dis[][i]=; 

    memset(dp,0x7f,sizeof(dp));
    dp[][][]=dp[][][]=;
    for(int i=;i<=n;i++){
        dp[i][][]=dp[i-][][]+dis[c[i-]][c[i]];
        for(int j=;j<=min(i,m);j++){
            int c1=c[i-],c2=c[i],d1=d[i-],d2=d[i];
            dp[i][j][]=min(dp[i][j][],
                        min(dp[i-][j][]+dis[c1][c2],
                            dp[i-][j][]+dis[d1][c2]*k[i-]+dis[c1][c2]*(1.0-k[i-])));

            dp[i][j][]=min(dp[i][j][],
                        min(dp[i-][j-][]+dis[c1][c2]*(1.0-k[i])+dis[c1][d2]*k[i],//上一次没换
                            dp[i-][j-][]+dis[d1][c2]*k[i-]*(1.0-k[i])//上一次换了成功且这一次没成功
                            +dis[d1][d2]*k[i-]*k[i]//上一次换了成功且这一次也成功
                            +dis[c1][d2]*(1.0-k[i-])*k[i]//上一次换了没成功且这一次成功
                            +dis[c1][c2]*(1.0-k[i-])*(1.0-k[i])));//上一次换了没成功且这一次也没成功
//            printf("%.2lf %.2lf\n",dp[i][j][0],dp[i][j][1]);
        }
    }

    ans=0x7fffffff;
    for(int i=;i<=m;i++)//从0开始，可以一次也不换
        ans=min(ans,min(dp[n][i][],dp[n][i][]));

    printf("%.2lf",ans);
    return ;
}