概率DP POJ3071

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;

double p[300][300];
double dp[10][300];

int main()
{
    int n;
    while(scanf("%d", &n) != EOF)
    {
        if ( n == -1 ) break;
        int m=1<<n;
        for(int i=0;i<m;i++)
            for(int t=0;t<m;t++)
                scanf("%lf",&p[i][t]);
        for(int i=0;i<m;i++)
            dp[0][i]=1.0;
        for(int i=1;i<=n;i++)
        {
            for(int t=0;t<m;t++)
            {
                dp[i][t]=0;
                for(int k=0;k<m;k++)
                {
                    if(((t>>(i-1))^1)==(k>>(i-1)))
                    {
                        dp[i][t]+=dp[i-1][t]*dp[i-1][k]*p[t][k];
                    }
                }
            }
        }
        int ans=0;
        for(int i=1;i<m;i++)
        {
            if(dp[n][i]>dp[n][ans])
                ans=i;
        }
        cout<<ans+1<<"
";
    }
    return 0;
}

最后注意只能用" "，用endl会刷新缓冲区，过不了