[概率dp] poj 3071 Football

题目分析

    队伍编号从0~2ⁿ，$dp_{ij}$代表进行i轮后，编号为j的队伍获胜的概率。
    $\begin{equation*} dp_{ij}= \sum_{k\in{S_{ij}}}dp_{(i-1)k}dp_{(i-1)k}a_{jk} \end{equation*}$，其中$S_{ij}$为第i轮j可能遇到的队伍的集合（这个概率dp其实就是模拟了比赛n轮）
    第i轮j可能遇到的队伍，是看队伍编号的二进制中2ⁱ位不同，2ⁱ⁺¹到2ⁿ位相同的，可以用位运算判断。

代码

#include <cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
using namespace std;
const int MAXN = (1<<7)+5;
int n,m;
double a[MAXN][MAXN];
double dp[MAXN][MAXN];//别用滚动数组
int main()
{
    while(scanf("%d",&n),~n)
    {
        m = n;
        n = (1<<n);
        for(int i = 0; i < n; i++)
            for(int j = 0; j < n; j++) scanf("%lf",&a[i][j]);
        for(int i = 0; i < n; i++) dp[0][i] = 1;
        for(int turn = 0; turn < m; turn++)
        {
            for(int i = 0; i < n; i++)
            {
                double sum = 0;
                for(int j = 0; j < n; j++)
                    if(((i>>turn)^1)==(j>>turn)) sum += dp[turn][j]*a[i][j];
                dp[turn+1][i] = dp[turn][i]*sum;
//                printf("turn:%3d %d %f \n",turn+1,i,dp[i]);
            }
        }
        double maxp=0;
        int ans;
        for(int i = 0; i < n; i++)
            if(maxp < dp[m][i])
            {
                maxp = dp[m][i];
                ans = i+1;
            }
//        cout<<dp[m][2]<<endl;
        cout<<ans<<endl;
    }
    return 0;
}

自己挖的坑

1. 不要用滚动数组，因为降维后在一轮中dp[i]要用到dp[j]，dp[j]也会用到dp[i]。
2. double maxp =0一开始忘记初始化！！！而且最开始用的int！！！要用浮点数以后千万不要再写成整型了啊啊啊啊啊啊啊啊啊啊啊！！！记得初始化局部变量啊啊啊啊啊啊！！！