HDU 4089 Activation【概率DP】

题目链接

题意：排队激活游戏，轮到队伍的第一个人有四种情况。1：连接失败，重新连接，仍然在第一个。2：激活失败，去队尾。3：激活成功，走掉。4：服务器崩溃。

现在有n个人，有个人排在第m个，求当服务器崩溃时他排在小于等于k个的概率。

令dp[i][j]表示i个人的队伍，排在第j个，服务器崩溃的概率。可以列出状态转移方程，然后化简。难点在于化简，因为i是从1到n循环的，所以到i的时候i-1的所有数都已经算出来了，相当于是常数。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
using namespace std;

int n, m, k;
double p1, p2, p3, p4;
double p21, p31, p41;
double dp[2010][2010];
double c[2010];
double pp[2010];

int main()
{
    while (scanf("%d %d %d %lf %lf %lf %lf", &n, &m, &k, &p1, &p2, &p3, &p4) != EOF)
    {
        if (fabs(p4) < 1e-8)
        {
            puts("0.00000");
            continue;
        }
        p21 = p2 / (1 - p1);
        p31 = p3 / (1 - p1);
        p41 = p4 / (1 - p1);
        pp[0] = 1;
        for (int i = 1; i <= n; i++)
        {
            if (i == 1)   pp[i] = p21;
            else    pp[i] = pp[i - 1] * p21;
        }
        c[1] = p41;
        dp[1][1] = c[1] / (1 - p21);
        for (int i = 2; i <= n; i++)
        {
            for (int j = 2; j <= i; j++)
            {
                if (j >= 2 && j <= k) c[j] = dp[i - 1][j - 1] * p31 + p41;
                if (j > k && j <= i)  c[j] = dp[i - 1][j - 1] * p31;
            }
            double tmp = 0.0;
            for (int j = 1; j <= i; j++)  tmp += c[j] * pp[i - j];
            dp[i][i] = tmp / (1 - pp[i]);
            dp[i][1] = p21 * dp[i][i] + c[1];
            for (int j = 2; j < i; j++)   dp[i][j] = p21 * dp[i][j - 1] + c[j];

        }
        printf("%.5f\n", dp[n][m]);
    }
    return 0;
}