HDU4089 Activation （概率dp）

题目链接：https://cn.vjudge.net/contest/247409#problem/D

题目大意：

有n个人排队等着在官网上激活游戏。Tomato排在第m个。 对于队列中的第一个人，有以下情况： 
1、激活失败，留在队列中等待下一次激活（概率为p1)
2、失去连接，出队列，然后排在队列的最后（概率为p2）
3、激活成功，离开队列（概率为p3） 
4、服务器瘫痪，服务器停止激活，所有人都无法激活了。
 求服务器瘫痪时Tomato在队列中的位置<=k的概率

题目思路：

用dp[i][j]记录队列里面有i个人，Tomato在第j个位置的概率，那最后dp[n][m]即为所求。

dp[i][1]=p1*dp[i][1]+dp[i][i]*p2+p4

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

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

令

p21=p2/(1-p1);

p31=p3/(1-p1);

p41=p4/(1-p1);

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

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

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

因为dp[i-1][j-1]在递推的时候可以解决

所以最后就转换成：

a[n]=An*a[n-1]+Cn;

a[n-1]=An-1 * a[n-2]+Cn-1;

...

a[2]=A2*a[1]+C2;

a[1]=A1*a[n]+C1;

通过迭代可以首先求出a[n],之后就可以求出所有的。

注意两大WA点：

必须特判下p4 接近 0的情况，不然计算代价奇高 
最后还的滚动数组优化，防止爆空间 

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string.h>
using namespace std;
const int maxn=2004;
double dp[2][maxn];
int n,m,k;
double p4,p1,p2,p3;
double p21,p31,p41;
double c[maxn];
int main()
{
    while(~scanf("%d%d%d",&n,&m,&k))
    {
        scanf("%lf%lf%lf%lf",&p1,&p2,&p3,&p4);
        if(p4<1e-5)
		{
			printf("0.00000\n");
			continue;
		}
		memset(dp,0,sizeof dp);
        p21=p2/(1-p1);
        p31=p3/(1-p1);
        p41=p4/(1-p1);
        dp[0][1]=p4/(1-p1-p2);
        double ans,sum;
        int i,j;
        int now=0;
        for(i=2;i<=n;i++)
        {
            now=now^1;
            for(j=i;j>=2;j--)
            {
                c[j]=p31*dp[now^1][j-1];
                if(j<=k)c[j]+=p41;
            }
            c[1]=p41;
            ans=1;
            sum=0;
            for(j=i;j>=1;j--)
            {
                sum+=ans*c[j];
                ans*=p21;
            }
            dp[now][i]=sum/(1-ans);
            dp[now][0]=dp[now][i];
            for(j=1;j<i;j++)
            {
                dp[now][j]=p21*dp[now][j-1]+c[j];
            }
        }
        printf("%.5f\n",dp[now][m]);
    }
    return 0;
}