NOIP2016D1T3-换教室

问题描述

对于刚上大学的牛牛来说，他面临的第一个问题是如何根据实际情况申请合适的课程。
在可以选择的课程中，有 2n 节课程安排在 n 个时间段上。在第 i （1≤i≤n ）个时间段上，两节内容相同的课程同时在不同的地点进行，其中，牛牛预先被安排在教室 ci上课，而另一节课程在教室di进行。
在不提交任何申请的情况下，学生们需要按时间段的顺序依次完成所有的 n 节安排好的课程。如果学生想更换第 i 节课程的教室，则需要提出申请。若申请通过，学生就可以在第 i 个时间段去教室di上课，否则仍然在教室ci上课。
由于更换教室的需求太多，申请不一定能获得通过。通过计算，牛牛发现申请更换第 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，表示有一条双向道路连接教室 a_j, b_j，通过这条道路需要耗费的体力值是 wj；保证 1≤aj,bj≤v，1≤wj≤100。
保证 1≤n≤2000 ，0≤m≤2000 ， 1≤v≤300 ， 0≤e≤90000 。
保证通过学校里的道路，从任何一间教室出发，都能到达其他所有的教室。
保证输入的实数最多包含 3 位小数
输出格式
输出一行，包含一个实数，四舍五入精确到小数点后恰好 2 位，表示答案。你的输出必须和标准输出完全一样才算正确。
测试数据保证四舍五入后的答案和准确答案的差的绝对值不大于 $4 \times 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

思路分析

一看到这句话

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

这不就是动态规划的无后效性吗？
那么我们开始设计出状态， 蒟蒻表示不会就先写个dp[i][j]吧 dp [i][j]就是在选择是否在当前这个时段换座位之前还剩余i个机会，目前已经选择到了j的最优方案。然后又不会了。。。
因为这样设计状态是没法转移的，因为处理当前走到的教室和上一次走到的教室是有关系的，你不知道上一次的教室就无法转移，因为我们设计的状态没有办法表示上一次的教室选择，所以很自然状态变为 dp[i][j][k] ， 前两维表示的意思没有改变第三维 k 表示处理到第 i 个教室，我没有没有选择在这个时间段换教室
然后就是这个最短路，点最多300，还要多源，于是我想的是一波floyd
最后我们写出dp方程，方程超级长。。。对只是长没有难的。

$$dp[j][i][0]=min(dp[j][i-1][0]+G[C[i-1]][C[i]],dp[j+1][i-1][1]+G[D[i-1]][C[i]]*k[i-1]+G[C[i-1]][C[i]]*(1-k[i-1]));$$
解释：上面是说这次没换教室的时候=min(前一个时间段不选的时候的dp加上前后的长度，前一个时间段选的时候的dp加上换教室成功后的长度 成功概率+不成功的路径不成功的概率)

$$dp[j][i][1]=min(dp[j][i-1][0]+G[C[i-1]][D[i]]*k[i]+G[C[i-1]][C[i]]*(1-k[i]),dp[j+1][i-1][1]+G[D[i-1]][D[i]]*k[i]*k[i-1]+G[C[i-1]][D[i]]*(1-k[i-1])*k[i]+G[D[i-1]][C[i]]*(1-k[i])*k[i-1]+G[C[i-1]][C[i]]*(1-k[i])*(1-k[i-1]));$$

解释：这个就有点复杂了，dp这次换教室的时候=min(前一个时间段不选的时候的dp+后一次换成功的概率成功长度+换不成功的概率不成功长度，前一个时间段选的时候的dp+前一个成功概率后一个成功概率前一个成功后一个成功的长度+前一个不成功概率后一个成功概率前一个不成功后一个成功的长度+前一个成功概率后一个不成功概率前一个成功后一个不成功的长度+前一个不成功概率后一个不成功概率前一个不成功后一个不成功的长度)
最后统一求一个min即可

AC代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
using namespace std;
const int INF=0x3f3f3f3f;
int G[320][320],C[2020],D[2002],m,n,v,e;
double dp[2020][2020][2],k[2020],ans;

void init(){    //初始化图和dp数组 
    for(int i=0;i<320;i++){
        for(int j=0;j<=320;j++){
            if(i==j)G[i][j]=0;
            else G[i][j]=INF;
        }
    }
    for(int i=0;i<2020;i++){
        for(int j=0;j<2020;j++){
            dp[i][j][0]=dp[i][j][1]=INF;
        }
    }
    ans=INF;
    return;
}

void floyd(){   //最短路 
    for(int a=1;a<=v;a++){
        for(int b=1;b<=v;b++){
            for(int c=1;c<=v;c++){
                if(G[b][a]+G[a][c]<G[b][c])G[b][c]=G[b][a]+G[a][c];
            }
        }
    }
    return;
}

int main(){
    init();
    cin>>n>>m>>v>>e;
    for(int i=1;i<=n;i++)cin>>C[i];     //输入 
    for(int i=1;i<=n;i++)cin>>D[i];
    for(int i=1;i<=n;i++)cin>>k[i];
    for(int i=1;i<=e;i++){int a,b,c;scanf("%d%d%d",&a,&b,&c);c=min(G[a][b],c);G[a][b]=c;G[b][a]=c;}
    floyd();
    for(int i=0;i<320;i++){G[i][i]=0;G[i][0]=0;G[0][i]=0;}  //清零边与对角线 
    dp[m][1][0] = dp[m][1][1] = 0;      //初始化dp 
    //dp[这次开始时剩余换教室机会][处理到第几个教室][这次换还是不换] 
    for(int i=2;i<=n;i++){
        dp[m][i][0] = dp[m][i-1][0]+G[C[i-1]][C[i]];    //初始化dp 
        for(int j=m;j>=max(m-i,0);j--){
            //这次换了教室了 
            dp[j][i][1]=min(dp[j][i-1][0]+G[C[i-1]][D[i]]*k[i]+G[C[i-1]][C[i]]*(1-k[i]),dp[j+1][i-1][1]+G[D[i-1]][D[i]]*k[i]*k[i-1]+G[C[i-1]][D[i]]*(1-k[i-1])*k[i]+G[D[i-1]][C[i]]*(1-k[i])*k[i-1]+G[C[i-1]][C[i]]*(1-k[i])*(1-k[i-1]));
            //这次没换教室 
            dp[j][i][0]=min(dp[j][i-1][0]+G[C[i-1]][C[i]],dp[j+1][i-1][1]+G[D[i-1]][C[i]]*k[i-1]+G[C[i-1]][C[i]]*(1-k[i-1]));
        }
    }
    for(int j=0;j<=m;j++){      //找小值 
        if(j==0)ans=min(ans,dp[0][n][0]);
        else ans=min(dp[j][n][0],min(dp[j][n][1],ans));
    }
    printf("%.2lf",ans); 
    return 0;
}