题意:一个软件有s个子系统,会产生n种bug,某个人一天发现一个bug,这个bug属于某种bug,发生在某个子系统里。求找到所有的n种bug和s个子系统所需的天数的数学期望。(程序中的bug数量几乎是无限的,因此在某个子系统中发现某个类别的新bug的概率在发现该子系统中的任何类别的bug之后不会减少,所以发现一个bug是某种bug的概率是1/n,属于某个子系统的概率是1/s)。
思路:dp[i][j]表示的是已经找到i个A,j个B距离所要达到的目标状态的期望值,然后将dp[i][j]分为4个状态
1、A和B都没有找到,-->对应的概率P1=i*j/(n*s);
2、找到一个A,-->对应的概率P2=(n-i)*j /(n*s);
3、找到一个B,-->对应的概率P3= i*(s-j) / (n*s);
4、找到一个A和B,-->对应的概率P4= (n-i)*(s-j) / (n*s);
根据数学期望公式得到:
dp[i,j] = p1*dp[i,j] + p2*dp[i+1,j] + p3*dp[i,j+1] + p4*dp[i+1,j+1] + 1;
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
double dp[1005][1005];
int main()
{
int n,s;
while(~scanf("%d%d",&n,&s))
{
memset(dp,0,sizeof(dp));
for(int i=n;i>=0;i--)
{
for(int j=s;j>=0;j--)
{
if(i!=n||j!=s)
{
double f=1.0*n*s;
double p=1+dp[i][j+1]*i*(s-j)/f+dp[i+1][j]*(n-i)*j/f+dp[i+1][j+1]*(n-i)*(s-j)/f;
double q=1-i*j/f;
dp[i][j]=p/q;
}
}
}
printf("%.4lf\n",dp[0][0]);
}
}