POJ 3071 Football

概率DP啊~~ 


题目大意:

有2^n个球队,每两个球队之间的获胜概率已经给出,问最后谁最有可能获胜。



解题思路:

dp[i][j]代表第i支球队大道第j场能获胜的概率。


下面是代码:

#include <set>
#include <map>
#include <queue>
#include <math.h>
#include <vector>
#include <string>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>

#define eps 1e-6
#define pi acos(-1.0)
#define inf 107374182
#define inf64 1152921504606846976
#define lc l,m,tr<<1
#define rc m + 1,r,tr<<1|1
#define iabs(x)  ((x) > 0 ? (x) : -(x))
#define clear1(A, X, SIZE) memset(A, X, sizeof(A[0]) * (SIZE))
#define clearall(A, X) memset(A, X, sizeof(A))
#define memcopy1(A , X, SIZE) memcpy(A , X ,sizeof(X[0])*(SIZE))
#define memcopyall(A, X) memcpy(A , X ,sizeof(X))
#define max( x, y )  ( ((x) > (y)) ? (x) : (y) )
#define min( x, y )  ( ((x) < (y)) ? (x) : (y) )

using namespace std;

double p[1<<7+5][1<<7+5];
double dp[10][1<<7];


int main()
{
    int T,n;
    while(scanf("%d",&T),T!=-1)
    {
        n=1<<T;
        for(int i=0; i<n; i++)
        {
            for(int j=0; j<n; j++)
            {
                scanf("%lf",&p[i][j]);
            }
        }
        for(int i=0; i<n; i+=2)
        {
            dp[1][i]=p[i][i+1];
            dp[1][i+1]=p[i+1][i];
        }
        double temp=0;
        for(int i=2; i<=T; i++)
        {
            for(int j=0; j<n; j++)
            {
                temp=0;
                dp[i][j]=dp[i-1][j];
                for(int k=0; k<n; k++)
                {
                    if(((k>>(i-1))^1)==(j>>(i-1)))
                    {
                        temp+=dp[i-1][k]*p[j][k];
                    }
                }
                dp[i][j]*=temp;
            }
        }
        int ans=0;
        for(int i=1; i<n; i++)
        {
            if(dp[T][i]>dp[T][ans])ans=i;
        }
        printf("%d\n",ans+1);
    }
    return 0;
}


你可能感兴趣的:(数学,poj,刷题)