题意:有n种卡片,你每买一包零食都有可能获得其中的一种卡片或者没获得卡片。问要集齐这n种卡片,需要买的零食包数的期望数是多少。
思路:状压dp+期望dp。因为最多有20种卡片,所以可以通过状态压缩表示获没获得该种卡片。0表示没有,1表示有,dp[i]表示状态i所需要买的零食期望数。最终集齐所有卡片的期望肯定是0,因为不需要再继续买了,因此dp[(1<<n)-1]=0,n为零食种数。dp[0]即为一张卡片都没有时需要买的包数的期望数,因此它就是答案。
可以发现,如果买的一包零食中没有卡片(概率为x),或者这包零食里的卡片已经拥有(即(i&1<<j)==1),那么该状态不会变。如果这包零食里的卡片没有((i&1<<j)==0),本状态才会改变,因此加上a[j]*dp[i|1<<j]。j为第j种零食,i为当前状态。
综合起来就是 dp[i]=x*(dp[i]+1)+∑a[k]*(dp[i]+1)+∑a[j]*(dp[i|1<<j]+1)
x表示买一包零食中没有卡片的概率
i为当前状态
j为第j种未拥有的卡片
k为第k种拥有的卡片
本来是做状压dp专题,没想到碰到了这个,顺便预习一下概率dp和期望dp。
AC代码:
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<string>
#include<queue>
#include<vector>
#include<map>
#include<set>
using namespace std;
const int mo=1e9;
const int mx=1<<20;
int n,m;
double a[mx];
int c[mx],ans,k;
int jud[mx];
double dp[mx];
int main() {
int t,i,j,l,q;
double x,y;
int cas=1,flag,f1;
char ch[mx];
while(scanf("%d",&n)!=EOF)
{
x=0;
for(i=0;i<n;i++) {scanf("%lf",&a[i]);
x+=a[i];
}
m=1<<n;
dp[m-1]=0;
x=1-x;
for(i=m-2;i>=0;i--)
{
dp[i]=1;
y=0;
for(j=0;j<n;j++)
{
if(i&1<<j) y+=a[j];
else dp[i]+=a[j]*dp[i|1<<j];
}
dp[i]/=(1-y-x);
}
printf("%5lf\n",dp[0]);
}
return 0;
}