ZOJ - 3551 Bloodsucker（期望dp）

最新推荐文章于 2021-04-11 21:06:41 发布

原创最新推荐文章于 2021-04-11 21:06:41 发布 · 323 阅读

0 ·

CC 4.0 BY-SA版权

动态规划同时被 2 个专栏收录

105 篇文章

订阅专栏

期望dp和期望概率问题

4 篇文章

订阅专栏

探讨一个由n个人组成的群体，其中包含1个吸血鬼和n-1个普通人。每天，两个随机个体相遇，若一方为吸血鬼，另一方为普通人，则普通人有概率m变为吸血鬼。本文通过动态规划求解所有个体都变为吸血鬼所需的平均天数。

In 0th day, there are n-1 people and 1 bloodsucker. Every day, two and only two of them meet. Nothing will happen if they are of the same species, that is, a people meets a people or a bloodsucker meets a bloodsucker. Otherwise, people may be transformed into bloodsucker with probability p. Sooner or later(D days), all people will be turned into bloodsucker. Calculate the mathematical expectation of D.

Input

The number of test cases (T, T ≤ 100) is given in the first line of the input. Each case consists of an integer n and a float number p (1 ≤ n < 100000, 0 < p ≤ 1, accurate to 3 digits after decimal point), separated by spaces.

Output

For each case, you should output the expectation(3 digits after the decimal point) in a single line.

Sample Input

1
2 1

Sample Output

1.000

题目大概：

给出n个人，有一个吸血鬼，n-1个普通人，每天会有两个人会相遇，如果一个是吸血鬼，一个是普通人，有m概率普通人会变成吸血鬼，问所有人都变成吸血鬼的期望天数。

思路：

dp[i]表示已经有i个吸血鬼，距离所有人变成吸血鬼还差的期望。
第i个人变成吸血鬼的概率可以计算出来为p=2*m*（n-i）*i/（n-1）/n。
可以得到一般方程dp[i]=(1-p)*dp[i]+p*dp[i+1]+1。
化简一下为dp[i]=dp[i+1]+1/p。

代码：

#include <bits/stdc++.h>

using namespace std;
const int maxn=1e5+10;

double dp[maxn];

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        memset(dp,0,sizeof(dp));
        int n;
        double m;
        scanf("%d%lf",&n,&m);
        dp[n]=0;
        for(int i=n-1;i>=1;i--)
        {
            double p=2.0*(double)i*(double)(n-i)/(double)n/(double)(n-1)*m;
            dp[i]=dp[i+1]+1/p;
        }
        printf("%.3lf\n",dp[1]);
    }
    return 0;
}