HDU4336题解(概率dp|容斥原理)

HDU4336题解

题意

小浣熊干脆面里装有英雄卡，有n种英雄卡，一包干脆面里最多有一张英雄卡，最少没有英雄卡，想知道在期望意义上买多少包干脆面可以把英雄卡集齐。

笺释

原理1

期望dp，在考虑期望dp的时候，要注意它和概率dp的本质都是递推。

这一点给出了用dp的思想解决统计学问题的理由，并且给出为什么我们不需要很高的统计学知识或者数论知识也能在acm中解决统计学和数论问题的理由:我们用到的只是穿上统计学外衣的dp原理，在实现方式与思考算法方面还是动态规划。

设dp[i]意为在状态i下还需要吃多少包干脆面(期望意义上)才能集齐卡片，状态i是二进制状压思想表示的，这里不多讲。
考虑dp[i]的转移方程。

一个原子操作(Atom Operation)显然是买一包干脆面，对于这一包干脆面来说，里面有可能没卡(P0)，有可能有卡但是是状态i中已经有的卡(P1)，有可能有卡并且是状态i中没有的卡(P2)。
只有第三种可能性能够改变状态i为i|(1《《j-1)。
因此，dp[i]=(P0+P1)*(dp[i]+1)+P2*dp[i|(1《《j-1)]+1）
至于为什么需要+1，可以这样疏解:因为是进行了一次原子操作得到的。

将dp[i]移项合并即可得到最终递推公式。

容斥原理做法

转载自https://blog.csdn.net/lianai911/article/details/47414619，讲得很棒。
思路：

Pi表示得到第i张卡牌的概率，Ei表示得到第i张卡的期望。

假设现在有两张卡牌，由题意可知：

E1 = 1/P1，E2 = 1/P2，E12(表示肯定买到1或2其中一包的期望) = 1/(P1+P2)。

当我们计算E1和E2的时候，E12是重复计算了2次，应该减去一次。根据容斥定理可知：

E = E1 + E2 - E12。

同理，三张牌的时候：

E = E1 + E2 + E3 - E12 - E13 - E23 + E123。

以此类推，当计算期望中的各项的时候，如果该项为奇数项(奇数张卡的期望)，则加上该项。

如果该项为偶数项2(偶数项卡的期望)，则减去该项。

完整代码

//DP
#include<bits/stdc++.h>
#define MAXN 21
using namespace std;
int n;
double P[MAXN],P0=1,dp[(1<<MAXN)-1];
int main()
{
    while(~scanf("%d",&n))
    {
        P0=1;
        for(int i=1;i<=n;i++)
        {
            scanf("%lf",&P[i]);
            P0-=P[i];
        }
        memset(dp,0,sizeof(dp));
        for(int i=(1<<n)-2;i>=0;i--)
        {
            double ans1=0,ans2=P0;
            for(int j=1;j<=n;j++)
            {
                if(i&(1<<(j-1)))
                {
                    ans2+=P[j];
                }
                else
                {
                    ans1+=P[j]*(1+dp[i|(1<<(j-1))]);
                }
            }
            dp[i]=(ans1+ans2)/(1-ans2);
        }
            printf("%.4f\n",dp[0]);
    }
}

//容斥原理

#include<bits/stdc++.h>
#define MAXN 21
using namespace std;
int n;
double P[MAXN];
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%lf",&P[i]);
        }
        double x=0;
        for(int i=1;i<=(1<<n)-1;i++)
        {
            double ans=0;
            int cnt=0;
            for(int j=1;j<=n;j++)
            {
                if(i&(1<<j-1))
                   {
                     //   printf("ss\n");
                       ans+=P[j];
                        cnt++;
                   }
            }
            if(cnt&1)
            {
                x+=1/ans;
            }
            else
            {
                x-=1/ans;
            }
        }
        printf("%.4f\n",x);
    }
}