lightoj 1050 - Marbles （概率DP）

来源：互联网发布：Linux修改jvm内存大小编辑：程序博客网时间：2024/05/20 00:12

思路：定义dp[i][j] 为袋子中有i个红球和j个红球时获胜的概率

那么根据题意我只可以任意拿而对手只拿蓝球。那么

dp[i][j] = （拿到红球的概率） * dp[i-1][j-1] + (拿到蓝球的概率) * dp[i][j-2];

边界：当红球没有时，获胜的概率为1

注意点：T比较大，需要把所有数据预处理出来直接查询，否则超时

/*********************************************** * Author: fisty * Created Time: 2015-08-23 10:40:51 * File Name   : 1050.cpp *********************************************** */#include <iostream>#include <cstring>#include <deque>#include <cmath>#include <queue>#include <stack>#include <list>#include <map>#include <set>#include <string>#include <vector>#include <cstdio>#include <bitset>#include <algorithm>using namespace std;#define Debug(x) cout << #x << " " << x <<endl#define Memset(x, a) memset(x, a, sizeof(x))const int INF = 0x3f3f3f3f;typedef long long LL;typedef pair<int, int> P;#define FOR(i, a, b) for(int i = a;i < b; i++)#define lson l, m, k<<1#define rson m+1, r, k<<1|1#define MAX_N 510int t;   int Red, Blue;double dp[MAX_N][MAX_N];double init(){    Memset(dp, 0);    for(int i = 1;i <= MAX_N; i++){        dp[0][i] = 1;    }    for(int i = 1;i < MAX_N; i++){        for(int j = 2;j < MAX_N; j++){            dp[i][j] = ((double)i / (double)(i + j) * dp[i-1][j-1]) + (j / (double)(i + j) * dp[i][j-2]);         }    }}int main() {    //freopen("in.cpp", "r", stdin);    //cin.tie(0);    //ios::sync_with_stdio(false);    int cnt = 1;    scanf("%d", &t);    init();    while(t--){        scanf("%d%d", &Red, &Blue);        printf("Case %d: ", cnt++);        double ans = dp[Red][Blue];        if( ans >= -1e-7 && ans <= 1e-7)            printf("0\n");        else        printf("%lf\n", ans);    }    return 0;}

0 0