UVA - 11181 数学

UVA - 11181

题意：

n个人去买东西，其中第i个人买东西的概率是p[i],最后只有r个人买了东西，求每个人实际买了东西的概率

代码：

//在r个人买东西的概率下每个人买了东西的概率，这是条件概率，因为最多20个人可以枚举所有的状态
//然后找到所有的r个人买东西的状态，算出总的概率，某个人在此条件下的概率就是这个人参与了的状态
//的概率和除以总概率。
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int sta[<<],n,r,nu;
double p[],sum[];
void init(){
    int N=(<<n);
    nu=;
    for(int i=;i<N;i++){
        int t=i,cnt=;
        while(t){
            cnt+=(t&);
            t>>=;
        }
        if(cnt==r) sta[++nu]=i;
    }
}
int main()
{
    int cas=;
    while(scanf("%d%d",&n,&r)==&&(n+r)){
        for(int i=;i<=n;i++)
            scanf("%lf",&p[i]);
        init();
        memset(sum,,sizeof(sum));
        double tot=;
        for(int i=;i<=nu;i++){
            int x=sta[i];
            double tmp=1.0;
            for(int j=;j<=n;j++){
                if(x&(<<(j-))) tmp*=p[j];
                else tmp*=(-p[j]);
            }
            tot+=tmp;
            for(int j=;j<=n;j++)
                if(x&(<<(j-))) sum[j]+=tmp;
        }
        printf("Case %d:\n",++cas);
        for(int i=;i<=n;i++)
            printf("%.6lf\n",sum[i]/tot);
    }
    return ;
}