概率dp—抽奖游戏

代码:

#include<bits/stdc++.h>
using namespace std;
double dp[2000][2000];
int n,m;
int main(){
  cin>>n>>m;
  for(int i = 0;i<=n;i++){
        for(int j = 0;j<=m;j++){
            if(i==0 && j==0)
                dp[i][j] = 0;
            else if(i==0)
                dp[i][j] = 0;
            else if(j==0)
                dp[i][j] = 1;
            else
                dp[i][j] = 1.0*i/(i+j) + 1.0*j/(i+j) * (j-1)/(i+j-1)* ( ( j-2>=0 ? 1.0*i/(i+j-2)*dp[i-1][j-2] : 0)  + ( j-3>=0 ? 1.0*(j-2)/(i+j-2)*dp[i][j-3] : 0 ));

         ///    cout<<"i:"<<i<<" j:"<<j<<" :"<<dp[i][j]<<endl;

        }
    }
    printf("%.4lf\n",dp[n][m]);
}
/*dp[n][m] =
1 中了 n/(m+n);
 (A抽了一张不中奖,剩下n张中奖,m-1张不中奖)  然后B也抽了一张不中奖
    剩下n张中奖,m-2张不中奖.B可以随便摸
              B如果摸到中奖,则剩下n-1和m-2   B如果摸到不中奖,则剩下n和m-3
2 没中 m/(n+m) * (m-1)/(n+m-1)*  {  dp[n-1][m-2]  + dp[n][m-3]     }
*/

该题源于360笔试题。

考试时推出公式,但是边界写错了。。。。真想学好dp。。。

发布了575 篇原创文章 · 获赞 111 · 访问量 12万+

猜你喜欢

转载自blog.csdn.net/xianpingping/article/details/105093964