需要集齐N张卡片,每买一包零食,里面可能有不同的卡片,存在每种卡片的概率已知,每包零食最多有一张卡片。问集齐N张的期望。
dp[i]表示i这个状态下还需要多少包的期望,dp[(1<<n)-1] = 0;求dp[0]逆推即可。
一开始不会写,看了别的状态转移方程:dp[i]=sum(dp[i|(1<<j)]*va[i]+1)/(1-sum(va[k])),其中i&(1<<j)==0,i&(1<<K)==1.刚开始看不理解。。。
如果想得到一张卡片,期望就是 1/va[k]。 上面的方程和这个应该是一个道理。因为i这个状态可以由很多状态得到 所以变成了sum(dp[i|(1<<j)]*va[i]+1),这个对应‘1’,然后得到这些卡片的概率为(1-sum(va[k])。所以状态转移方程就出来了。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 22;
const int inf = 1<<21;
int n;
double va[maxn];
double dp[1<<20];
int main()
{
while(scanf("%d",&n)!=EOF)
{
double w = 0;
for(int i=0;i<n;i++){
scanf("%lf",&va[i]);
w+=va[i];
}
w=1-w;
int bit = 1<<n;
dp[bit-1] = 0;
for(int i = bit-2;i>=0;i--)
{
double s1 = 0;
double s2 = 0;
for(int j=0;j<n;j++)
{
if(i&(1<<j)) s2 += va[j];
else s1 += dp[i|(1<<j)]*va[j];
}
s1 += 1;
dp[i] = s1/(1-s2-w);
}
printf("%.5lf\n",dp[0]);
}
return 0;
}