uva 11181 Probability|Given

条件概率 P(A|B) = P(AB) / P(B)

暴力dfs得到概率。

#include <iostream>
#include<stdio.h>
#include<cstring>
using namespace std;

double p[30],ans[30];
int n,r;
//i:当前第几个 sum:还能选几个 cur:当前已算出的概率
double dfs(int i,int sum,double cur)
{
    if(i > n )
    {
        if(sum == 0) {return cur;}
        else return 0;
    }
    double a = 0;
    if(sum >0)
    {
        a+=dfs(i+1,sum-1,cur*p[i]); //选择当前点
        ans[i]+=a;
    }
    a+=dfs(i+1,sum,cur*(1-p[i])); //不选当前点
    return a;
}

int main()
{
    int cnt = 1;
    while(scanf("%d %d",&n,&r),n+r)
    {
        memset(ans,0,sizeof ans);
        for(int i = 1; i <= n; i++) scanf("%lf",&p[i]);
        printf("Case %d:\n",cnt++);
        ans[0] = dfs(1,r,1);
        //ans[0]有r个人买东西的概率
        for(int i = 1; i <= n; i++)
        {
            printf("%.6lf\n",ans[i]/ans[0]);
        }
    }
    return 0;
}