题目:
给出N( N <= 100 ) 个数字,从第一个数字开始,每次获取当前位置的数字并随机行进1~6步,直到最后一个数字为止。求获取数字总和的期望值。
思路:
我们知道,对于一个点来说,他的期望值就是走到这一点后平均能够得到的数值,那么我们想要求一点的期望的时候,因为从1~6的概率是一样的,所以我们可以通过将其自身的数值和他后面六格以内的数值取平均后加和得来,状态设为走到这一点后可以得到的期望,状态转移方程就是
dp[ i ] = cave[ i ] + ( dp[ i + 1 ] + dp[ i + 2 ] + ... + dp[ i + 6 ] ) / 6
时间复杂度为O( n )
代码:
#include <bits/stdc++.h>
using namespace std;
double dp[ 101 ];
int cave[ 101 ];
int main( ) {
int T, N;
int i, j, k, t;
cin >> T;
for( t = 1; t <= T; t ++ ) {
cin >> N;
memset( dp, 0, sizeof( dp ) );
for ( i = 1; i <= N; i++ ) {
cin >> cave[ i ];
}
dp[ N ] = cave[ N ];
for ( i = N - 1 ; i >= 1; i-- ) {
k = min( N - i, 6 );
for ( j = 1; j <= k; j ++ ) {
dp[ i ] += cave[ i ] + dp[ i + j ];
}
dp[ i ] /= k;
}
printf( "Case %d: %.8f\n", t, dp[ 1 ] );
}
}