HDOJ 4336 Card Collector (状态压缩DP)

题意：每袋零食最多有 1 张卡片，卡片一共有 n 种(n ≤ 20)，每种出现的概率 Pi 给出。求集齐卡片所需零食袋数的期望。

分析：从数据量知道可能是状压，而且 dp 数组是一维的。

参考了题解。。。  dp[i] 表示已经集齐了集合 i 里的卡片，接下来全部集齐需要零食的期望。

公式：dp[i] = dp[i] * (P(none) + sigma(x 属于 i) (Px) + sigma(y 不属于 i)  (dp[i | (1<<y)] * Py)

如果 dp[i] 需要 dp[j] 来推出，则 i 一定是 j 的子集。

注意到如果 i 是 j 的子集，则一定有 i < j 。(我之前就没注意到这点。。。然后用队列做的比较麻烦还 TLE 了。。。)

所以只需要从大到小枚举子集即可（可用位运算加速枚举子集的办法）。

代码：

#include <cstdio>
#include <cstring>
using namespace std;

const int maxn = 20;
int n;
double p[maxn], dp[1<<maxn];
double pl;

int main()
{
    while(~scanf("%d", &n)) {
        int all = 0;
        pl = 1;
        for(int i = 0; i < n; i++) {
            all += (1<<i);
            scanf("%lf", &p[i]);
            pl -= p[i];
        }
        for(int cur = all; cur >= 0; cur = (cur-1)&all) {
            double p0 = pl;
            dp[cur] = 0;
            for(int i = 0; i < n; i++) {
                if(cur & (1<<i))
                    p0 += p[i];
                else dp[cur] += dp[cur | (1<<i)] * p[i];
            }
            if(cur != all) dp[cur] = (dp[cur] + 1) / (1 - p0);
            if(cur == 0) break;
        }
        printf("%lf\n", dp[0]);
    }

    return 0;
}