POJ 3071 Football（概率DP）

题目链接

http://poj.org/problem?id=3071

---

题解

题目大意：2^n个人比赛，每相邻的比一下决出胜者再相邻比，每两个人比拼有概率输赢，问最后谁获胜的概率最大。

明显比赛的关系是一棵二叉树，预处理出每个人j在第i轮获胜后的位置pos[i][j]。

设f[i][j]为第i轮j胜出的概率，枚举j在第i轮的对手k，则f[i][j] += f[i-1][k] * f[i-1][j] * p[j][k]，直接转移即可。（转移的条件是j,k在第i轮获胜后去的位置相同且第i-1轮不同，即pos[i][j] == pos[i][k] && pos[i-1][j] != pos[i-1][k]）

如何才能WA呢？多组数据不清空数组即可。

时间复杂度O(n*4^n*T)。

---

代码

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <cstdlib>
#define maxn 200

using namespace std;

int n, m;
double f[maxn][maxn];
double p[maxn][maxn];
int pos[maxn][maxn];

int main(){

    while(~ scanf("%d", &n) && (~ n)){

        m = 1;
        for(int i = 1; i <= n; i++)  m <<= 1;

        for(int i = 1; i <= m; i++)
            for(int j = 1; j <= m; j++)
                scanf("%lf", &p[i][j]);

        for(int i = 1; i <= m; i++){
            pos[0][i] = i;
            for(int j = 1; j <= n; j++)
                pos[j][i] = (pos[j-1][i] + 1) >> 1;
        }

        for(int i = 1; i <= m; i++)
            f[0][i] = 1.0;

        for(int i = 1; i <= n; i++){
            for(int j = 1; j <= m; j++){
                f[i][j] = 0.0;
                for(int k = 1; k <= m; k++){
                    if(pos[i][j] == pos[i][k] && pos[i-1][j] != pos[i-1][k])
                        f[i][j] += f[i-1][k] * f[i-1][j] * p[j][k];
                }
            }

        }

        double Max = f[n][1];
        int ans = 1;

        for(int i = 2; i <= m; i++){
            if(Max < f[n][i]){
                Max = f[n][i];
                ans = i;
            }
        }

        printf("%d\n", ans);
    }

    return 0;
}

---