uva-11181-条件概率

https://vjudge.net/problem/UVA-11181

　　　　给出n个人可能买东西的概率pi，已知有r个人买了东西，问每个人可能买东西的概率是多少。

　　设E:有r个人买了东西     Ei:第i个人买了东西

　　则有ans[i]=P(Ei|E)= P(Ei*E) / P(E)

　　对于P(E)我们可以枚举出所有有r个人购买的情况计算每种情况的概率然后累加，P(Ei*E)也能用这种方式得到，然后计算答案就好了。

#include<iostream>
#include<cstring>
#include<queue>
#include<cstdio>
#include<stack>
#include<set>
#include<map>
#include<cmath>
#include<ctime>
#include<time.h> 
#include<algorithm>
using namespace std;
#define mp make_pair
#define pb push_back
#define debug puts("debug")
#define LL long long 
#define pii pair<int,int>
#define eps 1e-12
int tot[1100000];
int main()
{
    int n,r,i,j,k,cas=0;
    double P[25];
    double E,ans[25];
    for(i=0;i<(1<<20);++i){
        int tmp=0;
        j=i;
        while(j){
            tot[i]+=(j&1);
            j>>=1;
        }
    }
    while(scanf("%d%d",&n,&r)==2&&(n||r)){
        E=0;
        memset(ans,0,sizeof(ans));
        for(i=1;i<=n;++i) scanf("%lf",P+i);
        int all=(1<<n);
        for(i=0;i<all;++i){
            if(tot[i]==r){
            double tmp=1;
                for(j=0;j<n;++j){
                    if(i&(1<<j)){
                        tmp*=P[j+1];
                    }
                    else{
                        tmp*=(1-P[j+1]);
                    }
                }
                
                for(j=0;j<n;++j)
                     if(i&(1<<j)) ans[j+1]+=tmp;
                E+=tmp;
            }
        }
        printf("Case %d:
",++cas);
        for(i=1;i<=n;++i) printf("%.6f
",ans[i]/E);
    }
    return 0; 
}