hdu4711Weather 概率dp

//第i个城市到第j个城市的概率ma[i][j]
//第i天的天气天气wet[i]
//第i个城市天气为j的概率wet_m[i][j]
//Hovey从0点开始，找出其概率最大的路线
//dp[i][j] 表示在第i天Hovey在第j个城市在所有路线的最大概率
//dp[i][j] = dp[i-1][k].p+ma[k][j]+wet_m[j][wet[i]]
//但是由于精度问题，需要取对数
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
using namespace std ;
const int maxn = 1010 ;
const double inf = 1e50;
const double eps = 1e-14 ;
struct node
{
    int pre ;
    double p ;
}dp[maxn][maxn] ;
double ma[maxn][maxn] ;
double wet_m[maxn][maxn] ;
int wet[maxn] ;
int ans[maxn] ;
int main()
{
    //freopen("in.txt" ,"r" ,stdin) ;
    int T ;
    scanf("%d" ,&T) ;
    int n , m , w ;
    while(T--)
    {
        scanf("%d%d%d", &n , &m , &w) ;
        for(int i = 1;i <= n;i++)
        scanf("%d" , &wet[i]) ;
        for(int i = 0;i < m;i++)
        for(int j = 0 ;j < m;j++)
        {
            scanf("%lf" , &ma[i][j]) ;
            if(ma[i][j]<=0)ma[i][j] = -inf ;
            else
            ma[i][j] = log(ma[i][j]) ;
        }
        for(int i = 0;i < m;i++)
        for(int j = 0;j < w;j++)
        {
            scanf("%lf" , &wet_m[i][j]) ;
            if(wet_m[i][j] <= 0)wet_m[i][j] = -inf ;
            else
            wet_m[i][j] = log(wet_m[i][j]);
        }
        //memset(dp , 0 , sizeof(dp)) ;
        for(int i = 0;i <= n;i++)
        for(int j = 0;j <= m;j++)
        dp[i][j].p = -inf ;
        dp[0][0].p = 0 ;
        for(int i = 1;i <= n;i++)
          for(int j = 0 ;j < m;j++)
           for(int k = 0;k < m ;k++)
           {
               double temp = dp[i-1][k].p+ma[k][j]+wet_m[j][wet[i]] ;
               if(temp > dp[i][j].p)
               {
                   dp[i][j].p = temp;
                   dp[i][j].pre = k ;
               }
           }
        double Max = -inf ;
        int pos = 0;
        for(int j = 0;j < m;j++)
        if(dp[n][j].p > Max+eps)
        {
            pos = j ;
            Max = dp[n][j].p ;
        }
        int  pr = pos ;
        for(int i = n ;i > 0 ;i--)
        {
            ans[i] = pr;
            pr = dp[i][pr].pre ;
        }
        for(int i = 1;i <= n;i++)
        printf("%d%c" , ans[i] , i == n ?'\n':' ') ;
    }


}