codeforces-148D-Bag of mice-확률 DP

1513 단어 codeforces
dp[x][y]: 현재 x개의 흰 쥐, y개의 검은 쥐, 공주가 이길 확률이 있습니다.
그러면:
만약 공주가 흰 쥐를 직접 얻게 된다면 확률은 x/(x+y)이고 공주가 이길 것이다.
만약에 공주가 검은 쥐를 잡았다고 가정하면 확률이 y/(x+y)이다. 그러면 공주가 이기려면 용이 검은 쥐를 가져야 한다. 확률은 (y-1)/(x+y-1)이다.
그러면 도망가는 쥐가 검은 쥐일 확률은 (y-2)/(x+y-2), 흰 쥐일 확률은 (x)/(x+y-2)이다.
그러면 dp[x][y]=x/(x+y)+y/(x+y)*(y-1)/(x+y-1)*(y-2)/(x+y-2)*dp[x][y-3]+(x)/(x+y-2)*dp[x-1][y-2]);
기억을 깊게 뒤져도 되고, DP를 직접 건네줘도 된다.
#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stdlib.h>
#include<vector>
#include<cmath>
#include<queue>
#include<set>
#include<stack>
using namespace std;
#define N 100010
#define LL long long
#define INF 0xfffffff
#define maxn 1100
double dp[maxn][maxn];
double dos(int x,int y)
{
    if(x<0||y<0)return 0;
    if(x==0)return 0;
    if(y==0)return 1;
    if(dp[x][y]>-0.5)return dp[x][y];
    double a,b;
    a=1.0*x;
    b=1.0*y;
    dp[x][y]=a/(a+b);
    if(x+y>=3)
    {
        dp[x][y]+=(b*(b-1)/((a+b)*(a+b-1)))*
        (dos(x-1,y-2)*a/(a+b-2)+dos(x,y-3)*(b-2)/(a+b-2));
    }
  //  cout<<x<<" "<<y<<" "<<dp[x][y]<<endl;
    return dp[x][y];
}
int main()
{
    int a,b;
    while(~scanf("%d%d",&a,&b))
    {
        memset(dp,-1,sizeof(dp));
        printf("%.10lf
",dos(a,b)); } return 0; }

좋은 웹페이지 즐겨찾기