ZOJ 3725 概率dp

太差劲了

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;

double dp[10010][150], p[150][150];

void init(int n)
{
    for(int i = 0 ; i < 10010 ; i++){
        for(int j = 0 ; j <= n ; j++){
            dp[i][j] = 0;
        }
    }
    for(int i = 1 ; i <= n ; i++){
        dp[0][i] = 1;
    }
}

int main(){
    int M;
    while(~scanf("%d",&M)){
        int n = M * (M - 1) * (M - 2) / 6;
        for(int i = 1 ; i <= n ; i++){
            for(int j = 1 ; j <= n ; j++){
                scanf("%lf",&p[i][j]);
            }
        }
        init(n);
        int Q;scanf("%d",&Q);
        for(int i = 1 ; i <= Q ; i++){
            int x;scanf("%d",&x);x++;//当前敌方队伍编号 
            for(int j = 1 ; j <= n ; j++){
                dp[i][j] = max(dp[i][j], dp[i - 1][j] * p[j][x]);
                dp[i][x] = max(dp[i][x], dp[i - 1][j] * p[j][x]);
            }
        }
        double res = 0;
        for(int i = 1 ; i <= n ; i++){
            res = max(res, dp[Q][i]);
        }
        printf("%.6lf
",res);
    }
    
    
    return 0;
}