POJ 3071 Football 【概率dp】

 1 #include<stdio.h>
 2 #include<string.h>
 3 #define mem(a, b) memset(a, b, sizeof(a))
 4 
 5 double p[150][150];
 6 double dp[10][150]; //代表第i轮第j个队伍存活的概率 
 7 
 8 int main()
 9 {
10     int n;
11     while(scanf("%d", &n) != EOF)
12     {
13         if(n == -1)
14             break;
15         mem(dp, 0);  //WA了一次在这里，忘了清0. 
16         int limit = 1 << n;//队伍编号处理为 0 ~ limit - 1 
17         for(int i = 0; i < limit; i ++)//比赛未开始前每个队伍存活概率都为1 
18             dp[0][i] = 1.0;
19         for(int i = 0; i < limit; i ++)
20         {
21             for(int j = 0; j < limit; j ++)
22             {
23                 scanf("%lf", &p[i][j]);
24             }
25         }
26         for(int i = 1; i <= n; i ++)//枚举第几轮 
27         {
28             for(int j = 0; j < limit; j ++)//枚举获胜队伍 
29             {
30                 for(int k = 0; k < limit; k ++)//枚举失败队伍 
31                 {
32                     if((j >> (i - 1) ^ 1) == k >> (i - 1) && j != k)//满足在同一轮会相遇,即邻接条件 
33                     {
34                         dp[i][j] += dp[i - 1][j] * dp[i - 1][k] * p[j][k];//两支比赛队伍上一轮必须都存活，并且在该轮j赢了k 
35                     }
36                 }
37             }
38         }
39         double maxx = -1.0;
40         int ans;
41         for(int i = 0; i < limit; i ++)
42             if(maxx < dp[n][i])
43             {
44                 maxx = dp[n][i];
45                 ans = i + 1;
46             }
47         printf("%d\n", ans);
48     }
49     return 0;
50 }