hdu #4336.Card Collector 题解

题目传送门

题目大意： 每次购买有 $a_i$ 的概率得到卡片 $i$，求期望购买几次得到所有卡片。

题解

做法1

好像还没有发现别人写这样的笨b做法。。

考虑一个顺推的dp，令 $f_S$ 表示已经拥有 $S$ 内的卡片，期望需要的步数；$h_S$ 表示从 $f_{\varnothing }$ 转移到 $f_S$ 的概率——你可以看做 所有 得到 $S$ 这些卡片的方案 的概率之和；$g_S={\sum }_{i\in S}{a}_i,p=1-{\sum }_{i=1}^n{a}_i$，$p$ 相当于购买一次啥都没有买个寂寞的概率。

考虑上一张新得到的卡片是谁，那么有转移：
$$f_S=\sum _{i\in S}\sum _{j=0}^{\infty }(f_{S\oplus i}+(j+1)\times h_{S\oplus i})\times (g_{S\oplus i}+p{)}^j\times a_i$$

令 $d=g_{S\oplus i}+p$，熟练使用生成函数的选手知道，当 $x\in (0,1)$ 时，${\sum }_{i=0}^{\infty }{x}^i=\frac{1}{1-x},{\sum }_{i=0}^{\infty }(i+1)x^i=\frac{1}{(1-x{)}^2}$，利用这两个公式推一推上面的式子：
$$\begin{array}{rl}{f}_S& =\sum _{i\in S}{f}_{S\oplus i}\times a_i\times \frac{1}{1-d}+h_{S\oplus i}\times a_i\frac{1}{(1-d{)}^2}\end{array}$$

这样就做完了，关于 $h$ 的递推就看代码吧：

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

int n,cnt[1<<20];double a[21];
double f[1<<20],g[1<<20],h[1<<20];

int main()
{
	for(int i=0;i<20;i++)cnt[1<<i]=i;
	while(~scanf("%d",&n))
	{
		double p=1;
		for(int i=0;i<n;i++)
			scanf("%lf",&a[i]),p-=a[i];
		for(int S=1;S<(1<<n);S++)
			g[S]=g[S-(S&-S)]+a[cnt[S&-S]],h[S]=0;
		h[0]=1;
		for(int S=0;S<(1<<n);S++){
			double sum=0;
			for(int i=0;i<n;i++)
				if(!(S>>i&1))sum+=a[i];
			for(int i=0;i<n;i++)
				if(!(S>>i&1))h[S|(1<<i)]+=h[S]*a[i]/sum;
				//a[i]/sum就是从S转移到S|(1<<i)的概率
		}
		for(int S=1;S<(1<<n);S++){
			f[S]=0;
			for(int i=0;i<n;i++)if(S>>i&1){
				double d=g[S^(1<<i)]+p;
				f[S]+=f[S^(1<<i)]*a[i]/(1-d)+1.0/(1-d)/(1-d)*a[i]*h[S^(1<<i)];
			}
		}
		printf("%.5lf\n",f[(1<<n)-1]);
	}
}

做法2

这种做法是考虑一种巧妙的倒推dp，也是网上除了做法 $3$ 之外的主流做法。

令 $f_S$ 表示已经拥有 $S$ 内的卡片，期望还需要多少次才能集齐所有卡片。

转移很简单，考虑一次购买得到了什么：
$$\begin{array}{rl}{f}_S& =1+(\sum _{i\notin S}{a}_i\times f_{S\oplus i})+(1-\sum _{i\notin S}{a}_i)f_S\\ f_S& =\frac{1+({\sum }_{i\notin S}{a}_i\times f_{S\oplus i})}{{\sum }_{i\notin S}{a}_i}\end{array}$$

代码也比上面的简单：

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

int n;
double a[20],f[1<<20];

int main()
{
	while(~scanf("%d",&n))
	{
		for(int i=0;i<n;i++)scanf("%lf",&a[i]);
		f[(1<<n)-1]=0;
		for(int S=(1<<n)-2;S>=0;S--){
			f[S]=1;double sum=0;
			for(int i=0;i<n;i++)if(!(S>>i&1))
				f[S]+=f[S|(1<<i)]*a[i],sum+=a[i];
			f[S]/=sum;
		}
		printf("%.5lf\n",f[0]);
	}
}

做法3

广为人知的Min-Max容斥做法，看例题 $1$。