hdu 4089 概率dp

带循环的概率dp

但是这个题目的问题是精度很难处理,为什么不特判p4就a不了,表示不能理解。

还有一个问题就是如果用维护部分和的办法,每个dp值都通过解方程得到,为什么就会tle,复杂度仍然是n*m的。

因为常数太大还是真的是因为运算过程出现了nan,inf之流的数据,使得运算过慢?

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=2e3+9;
int n,m,k;
double p1,p2,p3,p4;
double dp[maxn][maxn];

void solve()
{
    memset(dp,0,sizeof(dp));
    for(int i=1;i<=n;i++)
    {
        double sum=0,now=1;
        for(int j=i;j>=2;j--)
        {
            now/=(1-p1);
            if(j!=i)
            sum+=(p3*dp[i-1][j]+p4*(j+1<=k))*now;
            else
            sum+=(p4*(1<=k))*now;
            now*=p2;
        }
        int from=2;
        if(from>i) from=1;
        dp[i][1]=sum/(1-p2*(now/(1-p1)))+now/(1-p1)/(1-p2*(now/(1-p1)))*(p3*dp[i-1][1]+p4*(from<=k));
        for(int j=2;j<=i;j++)
        {
            dp[i][j]=p2/(1-p1)*dp[i][j-1]+p3/(1-p1)*dp[i-1][j-1]+p4/((1-p1))*(j<=k);
        }
    }
}

int main()
{
//    freopen("in.txt","r",stdin);
    while(scanf("%d%d%d",&n,&m,&k)!=EOF)
    {
        scanf("%lf%lf%lf%lf",&p1,&p2,&p3,&p4);
        solve();
        if(p4<1e-5)
        printf("0.00000\n");
        else
        printf("%.5f\n",dp[n][m]);
    }
    return 0;
}


你可能感兴趣的:(hdu 4089 概率dp)