【模拟试题】CARD COLLECTOR(BSOI4922)

【模拟试题】CARD COLLECTOR

Description

有N(N<=20)张卡片，现在你想集齐这N张卡片，所以你要购买一些卡包，每个卡包内可能没有卡片，可能有一张卡片。这N张卡片出现在卡包中的概率分别是p1,p2,p3……pn。求集齐N张卡片需要购买卡包数量的期望值。

Input

输入共两行，第一行一个整数N。
第二行N个整数，表示每张卡片出现概率。

Output

输出一行一个实数，结果保留5位小数。

Sample Input

2

0.1 0.4

Sample Output

10.50000

Hint

对于100%的数据，N<=20，p1+p2+p3+……+pn<=1.0且p1，p2，p3……pn>0.0。

Solution

 这道题的特殊之处就在于有空包（黑心商家），并且每张卡的概率不同（一辈子都抽不到ssr）。

因为最多只有20种不同的卡牌，所以我们可以联想到状压dp。设f[i]为I对应的情况。

若（i>>k）&1==1,则说明i情况下已经拿到了k号牌，那么f[i]由f[i-(1<<k)]转化来的概率为p[k]/(pi[1]+pi[2]+pi[3]+…L…pi[pn]).。

也就是说，这种情况的概率为k号牌的概率除以剩下牌的概率，只需将各种牌的情况累加即可。

CODE

#include<iostream>
#include<cstdio>
using namespace std;
int n;
double f[1<<22];
double p[25];
int main(){
	cin>>n;
	int i,j,k;
	int q=1;
	for(i=1;i<n;i++)q+=1<<i; 
	for(i=1;i<=q;i++)f[i]=0.0;
	for(i=1;i<=n;i++)cin>>p[i],f[1<<(i-1)]=1/p[i];
	double a,b;
	for(i=3;i<=q;i++){a=1.0;b=0.0;
		for(j=1;j<=n;j++){
			if((i>>(j-1))&1==1){
				a+=p[j]*f[i-(1<<(j-1))];
				b+=p[j];
			}
		}
		f[i]=a/b;
	}
	printf("%.5lf",f[q]);
	return 0;
}