题意:给出n种图案在每一张卡片上的出现的概率,一张卡片只能出现一种图案,或者不出现,求集齐所有图案所需的卡片期望数。
题解:期望+状压dp
二进制压缩,
d
p
[
i
]
dp[i]
dp[i]表示获得当前状态之后还需要的卡片数。
考虑两种情况。令p为当前状态未包含图案的概率和(Σp[j])。
①抽到已经有的图案或未抽到
dp[i] += (dp[i] + 1) * (1 - p)
②抽到未有的图案
dp[i] += (dp[i | 1 << (j - 1)] + 1) * p[j]
化简一下得dp[i]即可。
代码反过来写了,一样的。
#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<fstream>
#include<set>
#include<map>
#include<sstream>
#include<iomanip>
#define ll long long
using namespace std;
int n;
double p[22], dp[1111111];
int main() {
while (~scanf("%d", &n)) {
for (int i = 1; i <= n; i++) scanf("%lf", &p[i]);
for (int i = 1; i < (1 << n); i++) {
dp[i] = 1;
double sum = 0;
for (int j = 1; j <= n; j++) {
if (i & (1 << (j - 1))) {
dp[i] += dp[i - (1 << (j - 1))] * p[j];
sum += p[j];
}
}
dp[i] /= sum;
}
printf("%f\n", dp[(1 << n) - 1]);
}
return 0;
}