Help Me Escape ZOJ - 3640 && 概率dp

最新推荐文章于 2019-05-10 21:15:47 发布

deerly_

最新推荐文章于 2019-05-10 21:15:47 发布

阅读量187

点赞数

分类专栏： ---------dp--------- 概率dp -------期望-------

本文链接：https://blog.csdn.net/deerly_/article/details/82748371

版权

---------dp--------- 同时被 3 个专栏收录

45 篇文章 0 订阅

订阅专栏

-------期望-------

4 篇文章 0 订阅

订阅专栏

概率dp

3 篇文章 0 订阅

订阅专栏

该博客探讨了一道概率动态规划问题，即从具有不同防御力的洞口逃脱的期望天数。题目描述中，主角每次随机进入洞口，若战斗力超过洞口防御力，则按特定公式计算逃脱天数，否则战斗力增加。博主通过动态规划方法求解了期望值，并提供了关键思路和代码实现。

摘要由CSDN通过智能技术生成

题目大意

给你一个初始值f，有n个洞口，你每次被随机分到这n个洞口的其中一个，如果你的f>c[i]（洞的防御力），那么就可以跳出了，需要的天数是p*c[i]*c[i],如果f<=c[i]，那么他的攻击力变为f+c[i]，然后又随机到一个洞口，天数加1.求最后出去天数的期望值。
p = (1 + sqrt(5)) / 2. 向下取整

input

3 1
1 2 3

output

6.889

idea

dp[i]表示战斗力是i时，天数的期望。
当i > c[i]时，dp[i] += (p * c[i] * c[i]) / n;
当i <= c[i] 时， dp[i] += (dfs(c[i] + i) + 1) / n;
用到的思想就是期望就是概率乘上次数。

code

#include <bits/stdc++.h>
using namespace std;

const int maxn = 200005;
const double p = (1.0 + sqrt(5.0)) / 2;

int c[maxn], t[maxn], n;
double dp[maxn];

double dfs(int f) {
   if(dp[f] > 0) return dp[f];

   for(int i = 0; i < n; i++) {
      if(f > c[i]) dp[f] += (double)t[i] / n;
      else dp[f] += (double)(dfs(f + c[i]) + 1) / n;
   }

   return dp[f];
}

int main()
{
    int f;
    while(~scanf("%d %d", &n, &f)) {
        for(int i = 0; i < n; i++) {
        scanf("%d", &c[i]);
        t[i] = (int)((double)p * c[i] * c[i]);
    }

    memset(dp, 0, sizeof(dp));
    double ans = dfs(f);

    printf("%.3f\n", ans);
    }

}