LightOJ 1030 Discovering Gold 概率dp

/*
    题目描述：给出一个长度为n的洞穴，洞穴中每一处有一个价值为x[i]的宝藏，起初时位置在1处，现在用掷色子的方法决定
             向前走几步，如果向前走掷出的步数后超过n，那么就重新再掷一次，现在问走到n时获得宝藏值的数学期望是
             多少。
                    
    思路：设dp[i]表示从i位置走到n时获得的宝藏值的数学期望值，那么如果在i处掷一次色子保证不会跑到n之外的时候，到下一步
         每个位置的概率是1 / 6，那么dp[i] = Σ 1/6 *(x[i] + dp[i + j])    (1 <= j <= 6)，如果掷一次色子可能跑到n之外的时候，假
         设当前位置为i，则有i+6 > n，此时下一步走到任一点的概率p = 1 / ( n - i ) ，那么dp[i] = Σ p*(x[i] + dp[i + j])   (i + j <= 6)
         边界条件是dp[n] = x[n]
              
         以下是后一种情况概率为1 / (n - i)的证明：
         假设从i跳到n内某处k的概率为p，根据概率可知筛到可以跳出去的数的概率为 (6-(n - i))/6，那么有
         跳到k的概率 = 第一次跳到k的概率 + 第一次跳出去的概率 * 跳到k的概率
        翻译后 p = 1 / 6  + (6-(n - i))/6 * p    =>        p = 1 / ( n - i ) ;
*/
#pragma warning(disable:4786)
#pragma comment(linker, "/STACK:102400000,102400000")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<vector>
#include<cmath>
#include<string>
#include<sstream>
#define LL long long
#define FOR(i,f_start,f_end) for(int i=f_start;i<=f_end;++i)
#define mem(a,x) memset(a,x,sizeof(a))
#define lson l,m,x<<1
#define rson m+1,r,x<<1|1
using namespace std;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + 7;
const double PI = acos(-1.0);
const double eps=1e-8;
const int maxn = 1e2 + 5;
double dp[maxn] , x[maxn];
int main()
{
    int n , T , kase = 0 ;
    scanf("%d",&T);
    while(T--){
        mem(dp , 0);
        scanf("%d",&n);
        FOR( i , 1 , n )        scanf("%lf",&x[i]);
        dp[ n ] = x[ n ];
        for(int i = n - 1 ; i >= 1 ; i--){
            if( n - i < 6){
                double p = 1.0 / (n - i ) ;
                for(int j = i + 1 ; j <= n ; j++){
                    dp[i] += p*( x[i] + dp[j] );
                }
            }
            else{
                double p = 1.0 / 6.0 ;
                for(int j = i + 1 ; j<= i + 6 ; j++){
                    dp[i] += p * (x[i] + dp[j]);
                }
            }
        }
        printf("Case %d: %lf\n", ++kase , dp[1] ) ;
    }
    return 0;
}