题意:s个系统n种bug,每天找出一个bug,种类的概率是1/n,系统的概率是1/s。问:每个系统至少找出一个bug;每种类的bug都被找出。的期望天数(0<n, s<=1000)
#includeusing namespace std;double d[1005][1005];int n, s;double D;int main() { scanf("%d%d", &n, &s); d[s][n]=0; D=s*n; for(int i=s; i>=0; --i) for(int j=n; j>=0; --j) if(!(i==s&&j==n)) d[i][j]=((d[i+1][j]*(s-i)*j+d[i][j+1]*i*(n-j)+d[i+1][j+1]*(s-i)*(n-j))/D+1)/(1-i*j/D); printf("%.4f\n", d[0][0]); return 0;}
设$d(i, j)$表示已经找到了$i$个系统的bug,$j$种bug还需要的期望天数,显然答案是$d(0, 0)$
考虑转移:由于$d(i, j)$可以转移到5种子集,且互斥,转移如下:
- 权为1,概率为1。(表示今天找到的bug)
- 权为$d(i, j)$,概率为$i*j/s/n$。(表示由找到了一个旧系统且是旧种类的bug的状态转移过来)
- 权为$d(i+1, j)$,概率为$(s-i)*j/s/n$。(表示由找到了一个新系统但是是旧种类的bug的状态转移过来)
- 权为$d(i, j+1)$,概率为$i*(n-j)/s/n$。(表示由找到了一个旧系统但是是新种类的bug的状态转移过来)
- 权为$d(i+1, j+1)$,概率为$(s-i)*(n-j)/s/n$。(表示由找到了一个新系统且新种类的bug的状态转移过来)
发现是不是和一般的dp不同了呢?转移中有自己!哈哈,这就是期望dp的精髓所在= =
那么我们可以移项了= =(这样就不会无限递归,体现了数学的奥秘= =)最后得到:
$$d(i, j) = (ijd(i, j)+(s-i)jd(i-1, j)+i(n-j)d(i, j-1)+(s-i)(n-j)d(i-1, j-1)+1)/(n*s)/(1-ij/(n*s))$$
可能你会说,为什么要设“还需要的期望天数”而不是“转移到当前状态的期望天数”呢?因为在自己对自己的转移中,假设那样设状态的话,答案显然是$d(s, n)$对吧,可是你会发现转移方程除数变成0了!因此状态转移不过来...
而前者由于初始状态显然是$d(s, n)=0$,就不会出现除数为0也就是求极限的步骤啦= =,因此可以这样搞下去啦= =。(其实上边的状态不是不可做,可以无限递归,其实就是极限= =,换一种方式设状态就是避免了陷入求极限)