《算法竞赛进阶指南》218. 扑克牌 ·数学期望 + 记忆化搜索

题解

学习来源 - AcWing 218. 扑克牌

$E_{pre}=\sum (p_{next}\times E_{next})+cos{t}_{pre->next}$

数学期望标准套路 ↑

记：

• $a$ - 黑桃里被翻开的个数
• $b$ - 红桃里被翻开的个数
• $c$ - 梅花里被翻开的个数
• $d$ - 方块里被翻开的个数
• $e$ - 小王所在的位置
• $f$ - 大王所在的位置
• $sum$ - 被翻开的总数

$E_{a,b,c,d,e,f}=(\frac{13-a}{54-sum}\times E_{a+1,b,c,d,e,f}+\frac{13-b}{54-sum}\times E_{a,b+1,c,d,e,f}+\frac{13-c}{54-sum}\times E_{a,b,c+1,d,e,f}+\frac{13-d}{54-sum}\times E_{a,b,c,d+1,e,f}+\frac{1}{54-sum}\times E_{a,b,c,d,e=0,f}+\frac{1}{54-sum}\times E_{a+1,b,c,d,e,f=0})+1$

---

---

#include <bits/stdc++.h>
using namespace std;
const double eps = 1e-8;
int A, B, C, D;
double E[15][15][15][15][5][5];

// a 黑桃被翻开的个数
// b 红桃被翻开的个数
// c 梅花被翻开的个数
// d 方块被翻开的个数
// e 小王所在的位置
// f 大王所在的位置
double dfs(int a, int b, int c, int d, int e, int f) {
    if (E[a][b][c][d][e][f] > eps)
        return E[a][b][c][d][e][f];
    if (a + (e == 1) + (f == 1) >= A &&
        b + (e == 2) + (f == 2) >= B &&
        c + (e == 3) + (f == 3) >= C &&
        d + (e == 4) + (f == 4) >= D)
        return 0;

    int sum = a + b + c + d + (e != 0) + (f != 0);//已经抽出的总数
    double res = 1.0;//抽一张
    double Min1 = 1e9, Min2 = 1e9;
    if (a < 13) res += dfs(a + 1, b, c, d, e, f) * (13 - a) / (54 - sum);
    if (b < 13) res += dfs(a, b + 1, c, d, e, f) * (13 - b) / (54 - sum);
    if (c < 13) res += dfs(a, b, c + 1, d, e, f) * (13 - c) / (54 - sum);
    if (d < 13) res += dfs(a, b, c, d + 1, e, f) * (13 - d) / (54 - sum);
    if (e == 0) {
        for (int i = 1; i <= 4; ++i) {
            Min1 = min(Min1, dfs(a, b, c, d, i, f) / (54 - sum));
        }
        res += Min1;
    }
    if (f == 0) {
        for (int i = 1; i <= 4; ++i) {
            Min2 = min(Min2, dfs(a, b, c, d, e, i) / (54 - sum));
        }
        res += Min2;
    }
    return E[a][b][c][d][e][f] = res;
}

int main() {
    ios::sync_with_stdio(0);
    cin >> A >> B >> C >> D;
    if (max(A - 13, 0) + max(B - 13, 0) + max(C - 13, 0) + max(D - 13, 0) > 2) {
        cout << "-1.000" << endl;
    } else {
        printf("%.3f\n", dfs(0, 0, 0, 0, 0, 0));
    }
    return 0;
}