Codeforces 148D Bag of mice

链接

传送门

题意

\(w\)只白老鼠和\(b\)只黑老鼠,龙和公主轮流取,取到每只老鼠的概率相同,公主先取,龙取完之后会有一只老鼠跑掉。先取到白老鼠的胜利,如果没有人取到白老鼠,龙胜利。输出公主胜利的概率。

思路

定义状态\(d[i][j]\)为剩余\(i\)只白老鼠,\(j\)只黑老鼠是公主胜利的概率。只剩白老鼠时,概率为1;只剩黑老鼠时,概率为0。当公主和龙都取到黑老鼠时,可以根据跑掉的老鼠的颜色进行转移。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;

const int maxn = 1010;
double dp[maxn][maxn];
int main() {
int w, b;
scanf("%d%d", &w, &b);
for (int i = 0; i <= b; ++i) {
dp[0][i] = 0;
}
for (int i = 1; i <= w; ++i) {
dp[i][0] = 1;
}
for (int i = 1; i <= w; ++i) {
for (int j = 1; j <= b; ++j) {
dp[i][j] = double(i) / (i + j);
if (j >= 3) {
dp[i][j] += double(j) / (i + j) * double(j - 1) / (i + j - 1) * double(j - 2) / (i + j - 2) * dp[i][j - 3];
}
if (j >= 2) {
dp[i][j] += double(j) / (i + j) * double(j - 1) / (i + j - 1) * double(i) / (i + j - 2) * dp[i - 1][j - 2];
}
}
}
printf("%.10f\n", dp[w][b]); return 0; }