首页 > 代码库 > POJ 2096 (概率DP)

POJ 2096 (概率DP)

题目链接: http://poj.org/problem?id=2096

题目大意:n种bug,s个子系统。每天随机找一个bug,种类随机,来自系统随机。问找齐n种bug,且每个子系统至少有一个bug的期望天数。

解题思路

- -。题目像一坨屎。

其中"且每个子系统至少有一个bug"比较坑爹,其实意思就是找出s个bug就行了。

dp[i][j]表示已找到i种bug,且j个系统有bug的期望。

它可以由四个状态推到:

①dp[i][j], 当前找的bug,种类重复,且系统重复。概率为(i/n)*(j/s)。

②dp[i][j-1],当前找的bug,种类重复,且系统不重复。概率(i/n)*(s-j)/s。

③dp[i-1][j],当前找的bug,种类不重复,且系统重复。概率(n-i)/n*(j/s)。

④dp[i-1][j-1],当前找的bug,种类不重复,且系统不重复。概率为(n-i)/n*(s-j)/s。

最后dp[i][j]+=1.

由于是求期望,所以要逆推,dp[n][s]=0, ans=dp[0][0] 。

dp方程的减号全部改为加号。累加然后你会WA掉。因为double精度丢失太严重了。

dp[i][j]*(i/n)*(j/s)。几次乘几次除,精度会爆。

所以有必要进行化简, 把乘法化在一起,除法化在一起,最后做一步除法。

dp[i,j] = ( 1 + p2*dp[i+1,j] + p3*dp[i,j+1] + p4*dp[i+1,j+1] )/( 1-p1 )    
          = ( n*s + (n-i)*j*dp[i+1,j] + i*(s-j)*dp[i,j+1] + (n-i)*(s-j)*dp[i+1,j+1] )/( n*s - i*j )

由于POJ数据略水,其实化简到第一步就能A。

#include "cstdio"#include "cstring"double dp[1005][1005];int main(){    int n,s;    while(scanf("%d%d",&n,&s)!=EOF)    {        memset(dp,0,sizeof(dp));        for(int i=n;i>=0;i--)        {            for(int j=s;j>=0;j--)            {                if(i==n&&j==s) continue;                double p2=(double(s-j)*i)/n/s;                double p3=(double(n-i)*j)/n/s;                double p4=(double(n-i)*(s-j))/n/s;                double p1=1.0-(double(i*j))/n/s;                dp[i][j]=p2*dp[i][j+1]+p3*dp[i+1][j]+p4*dp[i+1][j+1]+1;                dp[i][j]/=p1;            }        }        printf("%.4lf\n",dp[0][0]);    }}

 

13634810neopenx2096Accepted8056K719MSC++679B2014-11-16 16:08:41

 

 

POJ 2096 (概率DP)