当前位置:编程学习 > C/C++ >>

hdu 4336 dp求期望(状态压缩)

题意:有n种卡片,吃零食的时候会吃到一些卡片,告诉你在一袋零食中吃到每种卡片

的概率,求搜集齐每种卡片所需要买零食的袋数的期望。

分析:n比较小,很自然的想到状态压缩DP

再分析一下转移过程的递推式就ok了

假设S状态中为1的数位表示还没有拿到的卡片,那么每次可能会
拿到这其中的某一张卡片,

也可能拿到原来已经拿到的卡片,

还可能一张卡片也拿不到

后两种情况的状态不变。
dp[0]=0;(表示每一种卡片都取完了,期望当然是0喽)
dp[S]=sum*dp[S]+p[x1]dp[S^(1<<x1)]+p[x2]dp[S^(1<<x2)].....+1;
sum是后两种情况的概率之和
移项,化简即可得到dp[S]的表达式
最后输出dp[(1<<n)-1]表示每一种卡片都没取时候的期望
[cpp] 
#include<cstdio> 
#include<cstring> 
double dp[1<<20]; 
double p[22]; 
int main() 

    int n; 
    while(scanf("%d",&n)!=EOF) 
    { 
        double x=0; 
        for(int i=0;i<n;i++) 
        { 
            scanf("%lf",&p[i]); 
            x+=p[i]; 
        } 
        dp[0]=0; 
        for(int i=1;i<(1<<n);i++) 
        { 
            double cnt=0; 
            double sum=0; 
            double s=0; 
            for(int j=0;j<n;j++) 
            { 
                if(!(i&(1<<j))) 
                { 
                    sum+=p[j]; 
                } 
                if(i&(1<<j)) 
                { 
                      s+=p[j]*dp[i^(1<<j)]; 
                } 
            } 
            sum+=1-x; 
            dp[i]=(s+1)/(1-sum); 
        } 
        printf("%.5lf\n",dp[(1<<n)-1]); 
    } 
    return 0;  


作者:haha593572013
补充:软件开发 , C++ ,
CopyRight © 2022 站长资源库 编程知识问答 zzzyk.com All Rights Reserved
部分文章来自网络,