UVa 11181 (条件概率) Probability|Given

题意：

有n个人买东西，第i个人买东西的概率为Pi。已知最终有r个人买了东西，求每个人买东西的概率。

分析：

设事件E为r个人买了东西，事件E_i为第i个人买了东西。所求为P(E_i|E) = P(E_iE) / P(E)

用一个buy数组记录每个人买或没买东西，然后dfs。

枚举所有r个人买了东西的情况的概率prob，累加到sum[n]中，对于buy[i] == true，再将prob加到sum[i]中。

最后答案为sum[i] / sum[n]

 1 #include <cstdio>
 2 #include <cstring>
 3 
 4 const int maxn = 25;
 5 int n, r;
 6 double P[maxn], sum[maxn];
 7 bool buy[maxn];
 8 
 9 void dfs(int d, int c, double prob)
10 {
11     if(c > r || d - c > n - r) return;  //有太多的1或者0
12     if(d == n)
13     {
14         sum[n] += prob;
15         for(int i = 0; i < n; ++i)
16             if(buy[i]) sum[i] += prob;
17         return;
18     }
19     buy[d] = 0;
20     dfs(d+1, c, prob * (1-P[d]));
21     buy[d] = 1;
22     dfs(d+1, c+1, prob * P[d]);
23 }
24 
25 int main()
26 {
27     //freopen("in.txt", "r", stdin);
28 
29     int kase = 0;
30     while(scanf("%d%d", &n, &r) == 2 && n)
31     {
32         for(int i = 0; i < n; ++i) scanf("%lf", &P[i]);
33         memset(sum, 0, sizeof(sum));
34         dfs(0, 0, 1.0);
35         printf("Case %d:\n", ++kase);
36         for(int i = 0; i < n; ++i)
37             printf("%.6f\n", sum[i] / sum[n]);
38     }
39 
40     return 0;
41 }

代码君

 

转载于:https://www.cnblogs.com/AOQNRMGYXLMV/p/4176208.html