洛谷P1850 换教室

题目描述

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

在可以选择的课程中，有 $c_ici 上课，而另一节课程在教室 d_idi 进行。$

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

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

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

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

牛牛所在的大学有

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

输入输出格式

输入格式：

第一行四个整数

第二行 $c_ici，即第 ii 个时间段牛牛被安排上课的教室；保证 1 le c_i le v1≤ci≤v。$

第三行 $d_idi，即第 ii 个时间段另一间上同样课程的教室；保证 1 le d_i le v1≤di≤v。$

第四行 $k_iki，即牛牛申请在第 ii 个时间段更换教室获得通过的概率。保证 0 le k_i le 10≤ki≤1。$

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

保证

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

保证输入的实数最多包含

输出格式：

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

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

输入输出样例

输入样例#1：

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

输出样例#1：

2.80

说明

【样例1说明】

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

【提示】

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

的是同一间教室。

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

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

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

483 055 310

题解：floyed+dp

代码:

68分暴力...开心......

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,m,v,e;
int c[20009],d[20009],dis[3002][3020];
double ans,k[20009];
int main(){
    scanf("%d%d%d%d",&n,&m,&v,&e);
    /*n表示时间段的数量，m为最多可以申请多少
    v表示学校里教室的数量，e表示学校里道路的数量。 
    */
    memset(dis,0x3f,sizeof(dis));
    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<=v;i++)dis[i][i]=0;
    for(int i=1;i<=e;i++){
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        dis[x][y]=dis[y][x]=min(dis[x][y],z);
    } 
    for(int kk=1;kk<=v;kk++)
     for(int i=1;i<=v;i++)
      for(int j=1;j<=v;j++)
       dis[i][j]=min(dis[i][j],dis[i][kk]+dis[kk][j]);
    if(m==0){
        for(int i=1;i<n;i++)ans+=dis[c[i]][c[i+1]]*1.0;
        printf("%.2lf",ans);
        return 0;
    }
    if(m==1){
        dis[0][d[1]]=0;dis[0][c[1]]=0;
        dis[d[n]][c[n+1]]=0;dis[c[n]][c[n+1]]=0;
        for(int i=1;i<n;i++)ans+=1.0*dis[c[i]][c[i+1]];
        double len=ans;
        for(int i=1;i<=n;i++){
            double tmp=0.0;
            tmp+=(1.0-k[i])*len;
            tmp+=k[i]*(len-dis[c[i-1]][c[i]]-dis[c[i]][c[i+1]]+dis[c[i-1]][d[i]]+dis[d[i]][c[i+1]]);
            ans=ans<tmp?ans:tmp; 
        }
        printf("%.2lf
",ans);
        return 0;
    }
    if(m==2){
        dis[0][d[1]]=0;dis[0][c[1]]=0;
        dis[d[n]][c[n+1]]=0;dis[c[n]][c[n+1]]=0;
        for(int i=1;i<n;i++)ans+=1.0*dis[c[i]][c[i+1]];
        double len=ans;
        for(int i=1;i<=n;i++){
            double tmp=0.0;
            tmp+=(1.0-k[i])*len;
            tmp+=k[i]*(len-dis[c[i-1]][c[i]]-dis[c[i]][c[i+1]]+dis[c[i-1]][d[i]]+dis[d[i]][c[i+1]]);
            ans=ans<tmp?ans:tmp; 
        }
        
        for(int i=1;i<=n;i++){
            for(int j=i+1;j<=n;j++){
                double tmp=0.0;
                tmp+=(1-k[i])*(1-k[j])*len;
                tmp+=k[i]*k[j]*(len-dis[c[i-1]][c[i]]-dis[c[i]][c[i+1]]+dis[c[i-1]][d[i]]+dis[d[i]][c[i+1]]-dis[c[j-1]][c[j]]-dis[c[j]][c[j+1]]+dis[c[j-1]][d[j]]+dis[d[j]][c[j+1]]);
                tmp+=(1-k[i])*k[j]*(len-dis[c[j-1]][c[j]]-dis[c[j]][c[j+1]]+dis[c[j-1]][d[j]]+dis[d[j]][c[j+1]]);
                tmp+=(1-k[j])*k[i]*(len-dis[c[i-1]][c[i]]-dis[c[i]][c[i+1]]+dis[c[i-1]][d[i]]+dis[d[i]][c[i+1]]);
                ans=ans<tmp?ans:tmp; 
            }
        }
        printf("%.2lf
",ans);
    }
    return 0;
}

正解：以后floyed还是循环赋值好了....免得把应该为0的设为Inf

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

int n,m,v,e;
int dis[2017][2017],c[2017],d[2017];
double k[2100],dp[2017][2017][3];

int main(){
    scanf("%d%d%d%d",&n,&m,&v,&e);
    /* 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<=v;i++){
    for(int j=1;j<=v;j++)
         dis[i][j]=0x3f3f3f3f;
         dis[i][i]=0;
    }
    for(int i=1;i<=v;i++)dis[i][i]=0;
    for(int i=1;i<=e;i++){
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        dis[x][y]=dis[y][x]=min(dis[x][y],z);
    }
    for(int kk=1;kk<=v;kk++)
     for(int i=1;i<=v;i++)
      for(int j=1;j<=v;j++)
       dis[i][j]=min(dis[i][j],dis[i][kk]+dis[kk][j]);
    for(int i=1;i<=n;i++){
        dp[i][0][0]=dp[i-1][0][0]+dis[c[i-1]][c[i]];
        dp[i][0][1]=1000000.0;
        for(int j=1;j<=m;j++){
            dp[i][j][0]=min(dp[i-1][j][0]+dis[c[i-1]][c[i]],
            dp[i-1][j][1]+k[i-1]*dis[d[i-1]][c[i]]+(1.0-k[i-1])*dis[c[i-1]][c[i]]);
            dp[i][j][1]=min(dp[i-1][j-1][0]+k[i]*dis[c[i-1]][d[i]]+(1.0-k[i])*dis[c[i-1]][c[i]],
            dp[i-1][j-1][1]+(1.0-k[i-1])*(1.0-k[i])*dis[c[i-1]][c[i]]+k[i]*k[i-1]*dis[d[i-1]][d[i]]+k[i-1]*(1.0-k[i])*dis[d[i-1]][c[i]]+(1.0-k[i-1])*k[i]*dis[c[i-1]][d[i]]
            );
        }
    }
    printf("%.2f",min(dp[n][m][0],dp[n][m][1]));
    return 0;
}