Light OJ 1030 - Discovering Gold(概率dp）

Posted 2020-08-22 aiterator

题目链接：http://www.lightoj.com/volume_showproblem.php?problem=1030

题目大意：有一个很长的洞穴， 可以看做是1-n的格子。你的起始位置在1的地方， 每个格子中都有价值为v[i]的宝藏。 有一个6面的骰子，数字为从1-6， 每次摇一次骰子， 得到的数字x后， 你可以到达距离当前位置大x的位置， 并且得到那个位置的宝藏。 如果要走的位置在n的外面， 那么在此摇骰子， 直到找到一个合适的数字。到达n位置的时候结束。 现在想知道走到n位置的能够获得宝藏价值的期望。

代码如下：从后往前推。

    #include<bits/stdc++.h>
    using namespace std;
     
    void solve(int _)
    {
        int n;
        scanf("%d", &n);
        int arr[105];
        for(int i=1; i<=n; ++ i)
            scanf("%d", &arr[i]);
        double dp[105];
        memset(dp, 0, sizeof(dp));
        for(int i=n; i >= 1; -- i)
        {
            dp[i] += arr[i];
            int t = min(n, i+6);
            for(int j=i+1; j<=t; ++ j)
                dp[i] += 1.0 / (t - i) * dp[j];
        }
        printf("Case %d: %.10f\\n", _, dp[1]);
    }
     
    int main()
    {
        int n;
        scanf("%d", &n);
        for(int i=1; i<=n; ++ i)
            solve(i);
        return 0;
    }
     

View Code