B - Collecting Bugs POJ - 概率DP

• B - Collecting Bugs

   POJ - 2096 
• 有n个bug1有s个bug2，ex[i][j]表示为已经发现了i个bug1，j个bug2,离目标发现s个bug1,n个bug2还需要的期望天数。这样所求就是ex[0][0]，而ex[n][s]则为0,因为已经全部发现，不需要额外的天数了。到下一天，产生三种发现新bug的情况，ex[i][j]，发现一个bug1, ex[i][j+1]，发现一个bug2，ex[i+1][j+1]，发现两个bug，那么
• ex[i][j]=（ 1+p2*ex[i+1][j]+p3*ex[i][j+1]+p4*ex[i+1][j+1] ）/（p1+p2+p3）  ，需要加1，因为是下一天、取均值

• #include <cstdio>
  #include <cstring>
  #include <iostream>
  #include <algorithm>
  using namespace std;
  const int N = 1e3 + 10 ;
  double f [N] [N] ;
  int main ()
  {
      int s, n ;
      while ( scanf ( "%d%d", & n, & s ) != EOF )
      {
          memset ( f, 0, sizeof f ) ;
          for ( int i = n ; i >= 0 ; -- i )
              for ( int j = s ; j >= 0 ; -- j )
              {
                  if ( i == n && j == s ) continue ;
                  double p2 = 1.0 * j / s * ( n - i ) / n ;
                  double p3 = 1.0 * ( s - j ) / s * i / n ;
                  double p4 = 1.0 * ( s - j ) / s * ( n - i ) / n ;
                  f [i] [j] = ( f [i+1] [j] * p2 + f [i] [j+1] * p3 + f [i+1] [j+1] * p4 + 1.0 ) / ( p2+p3+p4 ) ;
              }
          printf ( "%.5lf\n", f [0] [0] ) ;
      }
      return 0 ;
  }