NOIP 2016 换教室期望DP

最新推荐文章于 2025-06-15 18:04:51 发布

原创最新推荐文章于 2025-06-15 18:04:51 发布 · 539 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#期望DP

期望DP 同时被 2 个专栏收录

9 篇文章

订阅专栏

NOIP

2 篇文章

订阅专栏

本文介绍了一种通过动态规划解决教室更换路径优化问题的方法。利用三维数组存储状态，分别记录在考虑前i个教室，已经更换了j次教室的情况下，本次是否更换教室的期望距离。通过迭代更新状态，最终得到最小期望距离。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

传送门

令 $f[i][j][0]$ 表示当前枚举了 $i$ 个教室，换了 $k$ 次教室，这次不换的期望
令 $f[i][j][1]$ 表示当前枚举了 $i$ 个教室，换了 $k$ 次教室，这次换的期望

期望就是指一共只有前 $i$ 个教室，换 $j$ 个教室，当前换不换的期望距离
转移方程显然，见代码

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 2000 + 100;
const int INF = ( 1 << 29 ); 
int n,m,k,l, c[MAXN], d[MAXN], dis[MAXN][MAXN];
double f[MAXN][MAXN][2], p[MAXN], ans;

void init( ) {
    scanf( "%d%d%d%d", &n, &m, &k, &l );
    for( register int i = 1; i <= n; i++ ) scanf( "%d", &c[i] );
    for( register int i = 1; i <= n; i++ ) scanf( "%d", &d[i] );
    for( register int i = 1; i <= n; i++ ) scanf( "%lf", &p[i] );
    for( register int i = 1; i <= k; i++ ) for( register int j = 1; j <= k; j++ ) dis[i][j] = INF;
    for( register int i = 1; i <= l; i++ ){ int ff, tt, ww;
        scanf( "%d%d%d", &ff, &tt, &ww );
        if( dis[ff][tt] == INF ) dis[ff][tt] = dis[tt][ff] = ww;
        else                     dis[ff][tt] = min( dis[ff][tt], ww ), dis[tt][ff] = dis[ff][tt];
    }
    for( register int mid = 1; mid <= k; mid++ ) {
        for( register int i = 1; i <= k; i++ ) {
            if( i != mid ) {
                for( register int j = 1; j <= k; j++ ) {
                    if( j != i && j != mid ) {
                        dis[i][j] = min( dis[i][j], dis[i][mid] + dis[mid][j] );
                    } 
                }
            }
        }
    }
    for( register int i = 1; i <= n; i++ ) 
        for( register int j = 0; j <= m; j++ )
            f[i][j][0] = f[i][j][1] = 1e30;
    f[1][0][0] = f[1][1][1] = 0;//f[i][j][k]表示当前选了前i个,已经换了j次教室,当前这个教室换不换所期望的花费 
    for(int i=1;i<=k;i++) dis[i][i]=0;
    for( register int i = 1; i <= k; i++ ) dis[i][i] = 0;
    for( register int i = 2; i <= n; i++ ) {
        for( register int j = 0; 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.0 - p[ i - 1 ] ) + dis[ d[ i - 1 ] ][ c[i] ] * p[ i - 1 ] ) );
            if( j >= 1 ) { f[i][j][1] = min( f[i][j][1], min( f[ i - 1 ][ j - 1 ][0] + dis[ c[ i - 1 ] ][ d[i] ] * p[i] + dis[ c[ i - 1 ] ][ c[i] ] * ( 1.0 - p[i] ), f[ i - 1 ][ j - 1 ][1] + dis[ c[ i - 1 ] ][ c[i] ] * ( 1.0 - p[ i - 1 ] ) * ( 1.0 - p[i] ) + dis[ d[ i - 1 ] ][ c[i] ] * p[ i - 1 ] * ( 1.0 - p[i] ) + dis[ d[ i - 1 ] ][ d[i] ] * p[ i - 1 ] * p[i] + dis[ c[ i - 1 ] ][ d[ i ] ] * ( 1.0 - p[ i - 1 ] ) * p[i] ) ); } 
        }
    }
    ans = 1e30; for( register int i = 0; i <= m; i++ ) ans = min( ans, min( f[n][i][0], f[n][i][1] ) );
    printf("%.2lf",ans);
}
int main( ) {
    init();
    return 0;
}