UVA 11181（数学概率）

有n个人去超市，第i个人买东西的概率为p[i],逛完之后又r个人买了东西，求每个人实际买东西的概率。

条件概率公式的应用，具体分析见算法竞赛入门经典第二版p327

#include <stdio.h>
#include <string.h>
#include <cmath>
#include <algorithm>
using namespace std;
const int maxn = 100 + 10;
double p[maxn],a[maxn];
int n;
double dfs(int cur,int lft,double pnow)
{
if(lft==0){
if(cur==n+1)return pnow;
return dfs(cur+1,lft,pnow*(1-p[cur]));
}
if(n-cur+1<lft) return 0.0;
double res = 0;
double p1 = dfs(cur+1,lft-1,pnow*p[cur]);
double p2 = dfs(cur+1,lft,pnow*(1.0-p[cur]));
res = p1 + p2;
a[cur] += p1;
return res;
}
int main(int argc, char const *argv[])
{
int r;
for(int cas = 1; scanf("%d%d",&n,&r)==2; ++cas) {
memset(a,0,sizeof a);
if(n==0&&r==0)return 0;
for(int i = 1; i <= n; i++) {
scanf("%lf",p+i);
}
double tot = dfs(1,r,1.0);
printf("Case %d:\n", cas);
for(int i = 1; i <= n; i++)printf("%.6f\n", a[i]/tot);
}
return 0;
}