HDU 4089 Activation(11年北京 概率问题)

转载请注明出处，谢谢http://blog.csdn.net/acm_cxlove/article/details/7854526 by---cxlove

题目：仙5的激活序列。有以下4种情况：

1、注册失败，但是不影响队列顺序 ，概率为p1

2、连接失败，队首的人排到队尾，概率为p2

3、注册成功，队首离开队列，概率为p3

4、服务器崩溃，激活停止，概率为p4

求主角的位置在K以内，而且服务器崩溃的概率

http://acm.hdu.edu.cn/showproblem.php?pid=4089

比较容易想到一个二维DP

dp[i][j]表示队列总共有i个人，主角站在第j个位置时，所求事件发生的概率。

那么可以写出转移方程

dp[i][1]=dp[i][1]*p1+dp[i][i]*p2+p4 (可能自己注册失败，第二次继续，可能自己连接失败，回到队尾，服务器崩溃)

dp[i][j]=dp[i][j]*p1+dp[i][j-1]*p2+dp[i-1][j-1]*p3+p4 (j<=k)

dp[i][j]=dp[i][j]*p1+dp[i][j-1]*p2+dp[i-1][j-1]*p3 (j>k)

稍微整理下，把dp[i][j]放到一边

而且令 p21=p2/(1-p2) p31=p3/(1-p1) p41=p4/(1-p1);

dp[i][1]=dp[i][i]*p21+p41

dp[i][j]=dp[i][j-1]*p21+dp[i-1][j-1]*p31+p41; (j<=k)

dp[i][j]=dp[i][j-1]*p21+dp[i-1][j-1]*p31; (j>k)

可以发现后面的部分dp[i-1][j-1]在递推的时候都可以解决，我们令后面的部分为c[j]

则c[1]=p41 c[2]=dp[i-1][j-1]*p31+p41 …………

dp[i][j]就可以由dp[i][j-1]推出，但是可以发现这里产生 了一个循环，dp[i][1]又是由dp[i][i]得来的。

范围如此之大，肯定不能用高斯消元来做，而且也没必要。

我们把式子合并，迭代一下，这样就能先解出dp[i][i],进而可以得到dp[i][1],剩下的就可以递推出来了

dp[i][i]=dp[i][i]*p21^i+c[2]^p21^(i-1)……c[i]+p41*p21

题目要求的便是Dp[n][m];

#include<iostream>#include<cstdio>#include<algorithm>#include<cstring>#include<cmath>#define eps 1e-10#define N (1<<7)+5#define inf 1<<20#define zero(a) (fabs(a)<eps)#define lson (step<<1)#define rson (step<<1|1)using namespace std;double p1,p2,p3,p4,p21,p31,p41;double dp[2005][2005],c[2005];int n,m,k;int main(){    while(scanf("%d%d%d%lf%lf%lf%lf",&n,&m,&k,&p1,&p2,&p3,&p4)!=EOF){        if(zero(p4)) {puts("0.00000");continue;}        p21=p2/(1-p1);        p31=p3/(1-p1);        p41=p4/(1-p1);        dp[1][1]=p4/(1-p1-p2);        for(int i=2;i<=n;i++){            for(int j=2;j<=(i<k?i:k);j++)               c[j]=dp[i-1][j-1]*p31+p41;            for(int j=k+1;j<=i;j++)               c[j]=dp[i-1][j-1]*p31;            double p=1,tmp=0;            for(int j=i;j>1;j--){                tmp+=p*c[j];                p*=p21;            }            dp[i][i]=(tmp+p*p41)/(1-p*p21);            dp[i][1]=p21*dp[i][i]+p41;            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;}