H - Josephina and RPG

https://vjudge.net/contest/296454#problem/H

题目告诉你C(m,3)个队伍之间对战获胜的概率，然后挑战n个AI队伍，当你打败一支AI队伍的话可以选择和当前战胜的对手换队伍或者不换，求打败n支队伍的最大概率。

dp，，dp[i][j]表示用编号为j的队伍打败第i支AI队伍获胜的最大概率，，，p[i][j]:编号为i的队伍打败编号为j的队伍的概率

1：不换dp[i][j]=max(dp[i][j],d[i-1][j]*p[j][a[i]]);

2:  打完之后交换 dp[i][a[i]]=max(dp[i][a[i]],dp[i][j]*p[j][a[i]]);在编号为j的队伍打败第i支AI队伍后,交换队伍后队伍的状态变为dp[i][a[i]],更新dp[i][a[i]]的值

#include<stdio.h>
#include<string.h>
#include<string>
#include<queue>
#include<iostream>
#include<algorithm>
#include<map>
#include<cmath>
#include<set>
#include<math.h>
using namespace std;
typedef long long ll;
double dp[10005][150],p[150][150];
int a[10010];
int main()
{
    int m;
    while(~scanf("%d",&m))
    {
        memset(dp,0,sizeof(dp));
        m=m*(m-1)*(m-2)/6;///队伍总数C(m,3);
        for(int i=0; i<m; i++)
        {
            for(int j=0; j<m; j++)
                scanf("%lf",&p[i][j]);///储存概率
        }
        int n;
        scanf("%d",&n);
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);///储存AI队伍的信息
        }
        for(int i=0; i<m; i++)///打0个队伍，获胜的概率为1；
            dp[0][i]=1;
        for(int i=1; i<=n; i++)
        {
            for(int j=0; j<m; j++)
            {
                dp[i][j]=max(dp[i][j],dp[i-1][j]*p[j][a[i]]);///没有交换
                dp[i][a[i]]=max(dp[i][a[i]],dp[i-1][j]*p[j][a[i]]);///打完这一场后交换,状态变为dp[i][a[i-1]],更新dp[i][a[i-1]]的值
            }
        }
        double ans=0;
        for(int i=0; i<m; i++)///找出最大概率
            ans=max(ans,dp[n][i]);
        printf("%0.6f\n",ans);
    }
    return 0;
}