【BZOJ4720】【NOIP2016】换教室 [期望DP]

换教室

Time Limit: 20 Sec  Memory Limit: 512 MB
[Submit][Status][Discuss]

Description

Input

　　第一行四个整数n,m,v,e。n表示这个学期内的时间段的数量;m表示牛牛最多可以申请更换多少节课程的教室;

　　v表示牛牛学校里教室的数量;e表示牛牛的学校里道路的数量。

　　第二行n个正整数,第i(1≤i≤n)个正整数表示c,,即第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,表示有一条双向道路连接教室aj,bj,通过这条道路需要耗费的体力值是Wj;

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

　　保证输入的实数最多包含3位小数。

Output

　　输出一行,包含一个实数,四舎五入精确到小数点后恰好2位,表示答案。你的

　　输出必须和标准输出完全一样才算正确。

Sample Input

　　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

Sample Output

　　2.80

HINT

　　1≤aj,bj≤v, 1≤wj≤100。

　　1≤n≤2000, 0≤m≤2000, 1≤v≤300, 0≤e≤90000。

Main idea

　　给定n个 原本教室ci 和 替换教室di，可以申请m次换课，如果 i 换课了则可以在di上课，否则在ci上课，每个教室之间有距离，求期望最小距离。

Solution

　　很简单的期望DP，我们令 f[i][j][0\1] 表示 到了第 i 个状态，已经换了 j 次课，这次换不换课，然后分四种情况讨论一下即可。

Code

#include<iostream>
#include<string>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
using namespace std;  

#define Road(a,b) (double)w[a][b]
#define tense(a,b) a = a<b ? a:b;

const int ONE = ;
const double INF = 1e18;

int n,m,v,e;
int x,y,z;
int c[ONE],d[ONE];
int w[][];
double f[ONE][ONE][],k[ONE];
double Ans;

int get()
{
        int res=,Q=;char c;
        while( (c=getchar())< || c> )
        if(c=='-')Q=-;
        res=c-;
        while( (c=getchar())>= && c<= )
        res=res*+c-;
        return res*Q;
}

void Floyed()
{
        for(int k=; k<=v; k++)
        for(int i=; i<=v; i++)
        for(int j=; j<=v; j++)
            tense(w[i][j], w[i][k] + w[k][j]);
}

double Eap10(int i)
{
        return Road( c[i],c[i+] ) * (-k[i]) + Road( d[i],c[i+] ) * k[i];
}

double Eap01(int i)
{
        return Road( c[i],c[i+] ) * (-k[i+]) + Road( c[i],d[i+] ) * k[i+];
}

double Eap11(int i)
{
        return
            Road( c[i], c[i+] ) * (-k[i]) * (-k[i+])
          + Road( c[i], d[i+] ) * (-k[i]) * k[i+]
          + Road( d[i], c[i+] ) * k[i] * (-k[i+])
          + Road( d[i], d[i+] ) * k[i] * k[i+];
}

void DisApply(int i,int j)
{
        if(j>=) f[i+][j][] = tense(f[i+][j][], f[i][j][] + Road( c[i],c[i+] ) );
        if(j>=) f[i+][j][] = tense(f[i+][j][], f[i][j][] + Eap10(i) );
}

void Apply(int i,int j)
{
        if(j>=) f[i+][j][] = tense(f[i+][j][], f[i][j-][] + Eap01(i) );
        if(j>=) f[i+][j][] = tense(f[i+][j][], f[i][j-][] + Eap11(i) );
}

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

        memset(w, , sizeof(w));
        for(int i=; i<=v; i++) w[i][i] = ;
        for(int i=; i<=e; i++)
        {
            x = get();  y = get();  z = get();
            w[x][y] = min(w[x][y], z);
            w[y][x] = min(w[y][x], z);
        }
        Floyed();

        for(int i=; i<=n; i++)
            for(int j=; j<=m; j++)
                f[i][j][] = f[i][j][] = INF;
        f[][][] = f[][][] = ;

        for(int i=; i<=n-; i++)
            for(int j=; j<=m; j++)
                DisApply(i,j), Apply(i,j);

        Ans = INF;
        for(int j=; j<=m; j++)
        {
            tense(Ans, f[n][j][]);
            tense(Ans, f[n][j][]);
        }

        printf("%.2lf",Ans);
}