状态压缩+记忆化搜索：收集卡牌-csp202109-4

题目描述

csp202109-4

题目分析

中学数学题了属于是，其实没有什么数学上的概率知识难度，可以直接当模拟暴力去写。

暴力搜索20分

• 深度优先搜索。
• 有一个卡精度的问题就是，需要对每一层的结果统一乘以下一层的结果，而不是在递归出口直接算出结果累加。不然会在1e-8的精度被卡（题目的精度要求骗人！！！）

状态压缩+记忆化搜索

• 由于只有至多16张卡牌，学过状压的可能很容易想到用状压去表示卡牌持有状态（像我这样做了这题才知道什么是状压的另说……），也就是用一个int表示state，其中第i位表示是否持有第i张卡牌。
• 只压缩状态和暴力没有什么区别，需要记忆化搜索，目的是减少相同状态下的重复计算。
  • 使用二维数组dp记录某一level、某一state下对应的结果，state至多2^16，level至多n*k，数组不是很大的。

AC代码

#include <iostream>
using namespace std;

const int N_MAX = 20;
const int S_MAX = (1 << 16) + 10;
const int L_MAX = 5 * 20;
int n, k;
double p[N_MAX];    // probability of kps
double dp[L_MAX][S_MAX];

double dfs(int level, int state, int need, int coin) {    // need & coin可以被 level和state计算出，这里尝试逐层记录降低计算复杂度
    if (coin >= need * k || need == 0) return level;
    if (dp[level][state] != -1) return dp[level][state];
    double ans = 0;
    for (int i = 1; i <= n; i++) {
        if (state & (1 << (i-1))) ans += p[i] * dfs(level+1, state, need, coin+1);
        else ans += p[i] * dfs(level+1, state|(1<<(i-1)), need-1, coin);
    }
    return dp[level][state] = ans;
}

int main() {
    scanf("%d%d", &n, &k);
    for (int i = 1; i <= n; i++) scanf("%lf", &p[i]);
    for (int i = 0; i < L_MAX; i++) {
        for (int j = 0; j < S_MAX; j++) {
            dp[i][j] = -1;
        }
    }
    printf("%.10lf\n", dfs(0, 0, n, 0));        // 输出10位小数才能过，，，
    return 0;
}