LightOJ 1030 Discovering Gold（概率DP）题解

#include<set>
#include<map>
#include<stack>
#include<cmath>
#include<queue>
#include<vector>
#include<string>
#include<cstdio>
#include<cstring>
#include<sstream>
#include<iostream>
#include<algorithm>
typedef long long ll;
using namespace std;
const int maxn = 100 + 10;
const int MOD = 998244353;
const int INF = 0x3f3f3f3f;
int a[maxn];
double dp[maxn];    //从i开始走到n的期望
int main(){
    int n, ca = 1, t;
    scanf("%d", &t);
    while(t--){
        scanf("%d", &n);
        for(int i = 1; i <= n; i++)
            scanf("%d", &a[i]);
        dp[n] = a[n];
        for(int i = n - 1; i >= 1; i--){
            dp[i] = a[i];
            for(int j = i + 1; j <= min(i + 6, n); j++){
                dp[i] += 1.0 / min(6, n - i) * dp[j];
            }
        }
        printf("Case %d: %.8lf\n", ca++, dp[1]);
    }

    return 0;
}