zoj 3735 概率dp

链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3735
题意：
你有一只三个角色的队伍，需要打败所有电脑的队伍才能获胜，打赢一个之后，可以将你的队伍成员全部换成是被你打败的那个，当然也可以不换。
给了M个角色，R = C(M,3)表示队伍数，给了个矩阵表示每种队伍之间对战的胜率。
思路：
概率dp着实还是没有做过orz
一开始想到了dp，但是是从正着想的，发现不能满足没有后效性的。其实从后往前走就可以了。。
定义dp[i][j] 表示你用第j个的队伍打败 第i个电脑所用的队伍的最大的胜率。
t[i][j] 表示 i 打败 j 的胜率。 ai[i] 表示 第i个ai的队伍编号
可以发现就是两种状态，1)换队伍，2)不换队伍。
转移方程：dp[i][j] = t[j][ai[i]] * max(dp[i+1][j],dp[i+1][ai[i]])
最后dp[0]就是存的你用任一队伍打败最开始的电脑队伍的最大胜率，只要找出dp[0]中最大的一个概率就行了。
代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define MAXR 150
#define MAXN 10009
int m,r,n;
double t[MAXR][MAXR];
int ai[MAXN];
double dp[MAXN][MAXR];
int main()
{
    while(scanf("%d",&m) == 1)
    {
        r = m*(m-1)*(m-2)/3/2;
        for(int i = 0;i < r;i++)
        {
            for(int j = 0;j < r;j++)
                scanf("%lf",&t[i][j]);
        }
        scanf("%d",&n);
        for(int i = 0;i < n;i++)
            scanf("%d",&ai[i]);
        for(int i = 0;i <= n;i++)
        {
            for(int j = 0;j <= r;j++)
                dp[i][j] = 1;
        }
        for(int i = n-1;i >= 0;i--)
        {
            for(int j = 0;j < r;j++)
            {
                double p = t[j][ai[i]];
                dp[i][j] = p*max(dp[i+1][j],dp[i+1][ai[i]]);
            }
        }
        printf("%.6f\n",*max_element(dp[0],dp[0]+r));
    }
    return 0;
}