P1850 换教室【最短路】【概率DP】

题目描述

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

在可以选择的课程中，有 $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说明】

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

【提示】

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

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

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

慢慢调出来的，太恶心了，明天总结吧。。。

UPD:

#include <bits/stdc++.h>
#define dbg(x) cout << #x << "=" << x << endl
#define eps 1e-8
#define pi acos(-1.0)

using namespace std;
typedef long long LL;

const double inf = 1e17 + 7;

template<class T>inline void read(T &res)
{
    char c;T flag=1;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;res=c-'0';
    while((c=getchar())>='0'&&c<='9')res=res*10+c-'0';res*=flag;
}

namespace _buff {
    const size_t BUFF = 1 << 19;
    char ibuf[BUFF], *ib = ibuf, *ie = ibuf;
    char getc() {
        if (ib == ie) {
            ib = ibuf;
            ie = ibuf + fread(ibuf, 1, BUFF, stdin);
        }
        return ib == ie ? -1 : *ib++;
    }
}

int qread() {
    using namespace _buff;
    int ret = 0;
    bool pos = true;
    char c = getc();
    for (; (c < '0' || c > '9') && c != '-'; c = getc()) {
        assert(~c);
    }
    if (c == '-') {
        pos = false;
        c = getc();
    }
    for (; c >= '0' && c <= '9'; c = getc()) {
        ret = (ret << 3) + (ret << 1) + (c ^ 48);
    }
    return pos ? ret : -ret;
}

const int maxn = 2007;

int dis[maxn][maxn];
double f[maxn][maxn][2];
int c[maxn], d[maxn];
double k[maxn];

int n, m, v, e;

double Min(double a, double b) {
    if(a < b) {
        return a;
    }
    return b;
}

int main()
{
    //freopen("data.txt", "r",stdin);
    scanf("%d %d %d %d",&n, &m, &v, &e);
    for ( int i = 1; i <= n; ++i ) {
        read(c[i]);
    }   
    for ( int i = 1; i <= n; ++i ) {
        read(d[i]);
    }
    for ( int i = 1; i <= n; ++i ) {
        scanf("%lf",&k[i]);
    }
    memset(dis, 63, sizeof(dis));
    for ( int i = 1; i <= e; ++i ) {
        int u, v, w;
        scanf("%d %d %d",&u, &v, &w);
        dis[u][v] = dis[v][u] = min(w, dis[u][v]);
    }
    for ( int k = 1; k <= v; ++k ) {
        for ( int i = 1; i <= v; ++i ) {
            for ( int j = 1; j <= v; ++j ) {
                dis[i][j] = min(dis[i][j], dis[i][k] + dis[k][j]);
            }
        }
    }
    for ( int i = 1; i <= v; ++i ) {
        dis[i][i] = dis[i][0] = dis[0][i] = 0;
    }
    for ( int i = 0; i <= n; ++i ) {
        for ( int j = 0; j <= m; ++j ) {
            f[i][j][0] = inf;
            f[i][j][1] = inf;
        }
    }
    f[1][0][0] = f[1][1][1] = 0;
    for (register int i = 2; i <= n; i++) {
        f[i][0][0] = f[i - 1][0][0] + dis[c[i - 1]][c[i]];
        for (register int j = 1; j <= min(i, m); j++) {
            f[i][j][0] = Min(f[i][j][0], Min(f[i - 1][j][0] + dis[c[i-1]][c[i]], f[i - 1][j][1] + dis[c[i-1]][c[i]] * (1 - k[i - 1]) + dis[d[i - 1]][c[i]] * k[i - 1]));
            f[i][j][1] = Min(f[i][j][1], Min(f[i - 1][j - 1][0] + dis[c[i-1]][c[i]] * (1 - k[i]) + dis[c[i - 1]][d[i]] * k[i], f[i - 1][j - 1][1] + dis[d[i - 1]][d[i]] * k[i] * k[i - 1] + dis[d[i - 1]][c[i]] * k[i - 1] * (1 - k[i]) + dis[c[i - 1]][d[i]] * (1 - k[i - 1]) * k[i] + dis[c[i - 1]][c[i]] * (1 - k[i - 1]) * (1 - k[i])));
        }
    }
    double ans = inf;
    for ( int i = 0; i <= m; ++i ) {
        ans = Min(ans, Min(f[n][i][0], f[n][i][1]));
    }
    printf("%.2f
",ans);
    return 0;
}