Codeforces Round #301 (Div. 2)---D. Bad Luck Island(概率dp)

The Bad Luck Island is inhabited by three kinds of species: r rocks, s scissors and p papers. At some moments of time two random individuals meet (all pairs of individuals can meet equiprobably), and if they belong to different species, then one individual kills the other one: a rock kills scissors, scissors kill paper, and paper kills a rock. Your task is to determine for each species what is the probability that this species will be the only one to inhabit this island after a long enough period of time.
Input

The single line contains three integers r, s and p (1 ≤ r, s, p ≤ 100) — the original number of individuals in the species of rock, scissors and paper, respectively.
Output

Print three space-separated real numbers: the probabilities, at which the rocks, the scissors and the paper will be the only surviving species, respectively. The answer will be considered correct if the relative or absolute error of each number doesn’t exceed 10 - 9.
Sample test(s)
Input

2 2 2

Output

0.333333333333 0.333333333333 0.333333333333

Input

2 1 2

Output

0.150000000000 0.300000000000 0.550000000000

Input

1 1 3

Output

0.057142857143 0.657142857143 0.285714285714

dp[r][s][p]表示剩r个rock,s个scissors,p个papers的概率

/************************************************************************* > File Name: d.cpp > Author: ALex > Mail: [email protected] > Created Time: 2015年05月01日 星期五 20时06分20秒 ************************************************************************/

#include <functional>
#include <algorithm>
#include <iostream>
#include <fstream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <queue>
#include <stack>
#include <map>
#include <bitset>
#include <set>
#include <vector>

using namespace std;

const double pi = acos(-1.0);
const int inf = 0x3f3f3f3f;
const double eps = 1e-15;
typedef long long LL;
typedef pair <int, int> PLL;

double dp[101][101][101];

int main() {
    int r, s, p;
    while (cin >> r >> s >> p) {
        for (int i = 0; i <= r; ++i) {
            for (int j = 0; j <= s; ++j) {
                for (int k = 0; k <= p; ++k) {
                    dp[i][j][k] = 0;
                }
            }
        }
        dp[r][s][p] = 1;
        for (int i = r; i >= 0; --i) {
            for (int j = s; j >= 0; --j) {
                for (int k = p; k >= 0; --k) {
                    double pp = 0;
                    if (i + j + k < 1) {
                        continue;
                    }
                    if (i > 1) {
                        pp += i * (i - 1) * 1.0 / 2;
                    }
                    if (j > 1) {
                        pp += j * (j - 1) * 1.0 / 2;
                    }
                    if (k > 1) {
                        pp += k * (k - 1) * 1.0 / 2;
                    }
                    double x = ((i + j + k + 1) * (i + j + k) * 1.0) / 2;
                    if (i + j + k > 1) {
                        pp /= ((i + j + k) * (i + j + k - 1)) / 2;
                    }
                    else {
                        pp = 0;
                    }
                    double tmp = 0;
                    if (i + 1 <= r && k > 0) {
                        dp[i][j][k] += dp[i + 1][j][k] * ((i + 1) * k * 1.0) / x;
                    }
                    if (j + 1 <= s && i > 0) {
                        dp[i][j][k] += dp[i][j + 1][k] * ((j + 1) * i * 1.0) / x;
                    }
                    if (k + 1 <= p && j > 0) {
                        dp[i][j][k] += dp[i][j][k + 1] * ((k + 1) * j * 1.0) / x;
                    }
                    if (pp != 1.0) {
                        dp[i][j][k] /= (1.0 - pp);
                    }
            // dp[i][j][k] += tmp;
            // printf("dp[%d][%d][%d] = %lf\n", i, j, k, dp[i][j][k]);
                }
            }
        }
        double ans1 = 0, ans2 = 0, ans3 = 0;
        for (int i = 1; i <= r; ++i) {
            ans1 += dp[i][0][0];
        }
        for (int i = 1; i <= s; ++i) {
            ans2 += dp[0][i][0];
        }
        for (int i = 1; i <= p; ++i) {
            ans3 += dp[0][0][i];
        }
        printf("%.12f %.12f %.12f\n", ans1, ans2, ans3);
    }
    return 0;
}

你可能感兴趣的:(dp)