一道概率问题的编程解

问题：投掷21次硬币，至少一次连续5次或5次以上正面朝上的概率是多少？

我的解答：（brute函数是水木上另一个人给出的解）

1. #include <cstdio>
2. #include <cstdlib>
3. #include <memory.h>
4. #include <cmath>
5. #include <vector>
6. using namespace std;
8. template<int n, int m> int brute()
9. {
10.         int c = 0;
11.         for(unsigned int i = 0; i < (1 << n); i++)
12.         {
13.                 for(unsigned int t = i; t != 0; t >>= 1)
14.                 {
15.                         if( (t & ((1 << m) - 1)) == ((1 << m) - 1))
16.                         {
17.                                 c++;
18.                                 break;
19.                         }
20.                 }
21.         }
22.         return c;
23. }
25. template<int n, int m> double probability()
26. {
27.         int c = brute<n, m>();
28.         return (double)c / (1 << n);
29. }
30. // continuous M from N element
31. float getPossibility(int N, int M) {
32.   int n = M-1;
33.   vector< vector<int> > f(2, vector<int>(M,0));
34.   
35.   for (int i = 0; i < M-1; ++i) {
36.     f[0][i] = 1<<(M-i-2);
37.   }
38.   f[0][M-1] = 1;
40.   int cur = 0;
42.   while (n < N) {
43.     n++;
44.     int last = cur;
45.     cur = (cur+1)%2;
47.     int sum = f[last][4];
48.     for (int cnt = 1; cnt < M; ++cnt) {
49.       f[cur][cnt] = f[last][cnt-1];
50.       sum += f[last][cnt-1];
51.     }
52.     f[cur][0] = sum;
53.   }
54.   int ret = 0;
55.   for (int cnt = 0; cnt < M; ++cnt) {
56.     ret += f[cur][cnt];
57.   }
58.   return 1.0-((float)ret)/(1<<N);
59. }
60. int main()
61. {
62.         float p = getPossibility(5, 5);
63.         printf("%d, %f, %f/n", brute<5, 5>(), probability<5, 5>(), p);
64.         p = getPossibility(6, 5);
65.         printf("%d, %f, %f/n", brute<6, 5>(), probability<6, 5>(), p );
66.         p = getPossibility(21, 5);
67.         printf("%d, %f, %f/n", brute<21, 5>(), probability<21, 5>(), p );
68.         return 0;
69. }

简单的说明：计算所有5次以下的事件的个数。计算过程可用动态规划进行。