hdu5781[ATM Mechine] 浅尝期望概率DP

Venishel

于 2017-10-30 20:07:12 发布

阅读量215

点赞数

文章标签：期望概率DP

本文链接：https://blog.csdn.net/qq_37321281/article/details/78396780

版权

E ( i , j ) 表示存款在 [ 0 , i ] 这个区间，还能被警告 j 次的期望

E ( i , j ) = min { E ( i - k , j ) * P1 + E ( k - 1 , j - 1 ) * P2 + 1 }

P1 = ( i + 1 - k ) / ( i + 1 ) , P2 = k / ( i + 1 )

P1表示 i >= k 的概率，如果 i >= k , 那么 i 可以减掉 k

P2表示 i < k 的概率，如果 i < k , 那么被警告一次，i 可以取 [ 0 , k-1 ] 之间的数

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm> 
using namespace std;
const int N = 2010 ;
const int M = 20 ;
const int Inf = 1e9 ;

double E [N] [M] ;

int main () {
    for ( int i = 1 ; i <= 2000 ; ++ i ) E [i] [0] = ( double ) Inf ;
    for ( int i = 0 ; i <= 15 ; ++ i ) E [0] [i] = ( double ) 0 ;
    for ( int i = 1 ; i <= 2000 ; ++ i ) {
        for ( int j = 1 ; j <= 15 ; ++ j ){
            E [i] [j] = ( double ) Inf ;
            for ( int k = 0 ; k <= i ; ++ k )
                E [i] [j] = min ( E [i] [j] , E [i-k] [j] * ( i - k + 1.0 ) / ( i + 1.0 ) + E [ k - 1 ] [ j - 1 ] * ( k / ( i + 1.0 ) ) + 1 ) ;     
        }
    }
    int n , w ;
    while ( scanf ( "%d%d" , & n , & w ) != EOF ) {
        printf ( "%.6lf\n" , E [n] [ min ( 15 , w ) ] ) ;
    }
    return 0 ;
}