uva10417 概率DP

这题 to[i][j] 为第i个人送j这个礼物的概率 我们用13进制进行压缩这个留下的的礼物的个数，这样我们将dp[i][k]表示为当第i个人放完礼物后得到的状态为k时的概率，那么通过记忆化搜索我们就得到了我们想要的状态概率即 dp[i][k]+={dp[i+1][k-aj]*to[i][aj]}，好那么现在如果我去拿第一个礼物我正确的概率，我们 可以先知道好朋友放第一个的概率为

假设S为最初礼物的状态

pop=dp[1][S-a1]*to[0][1]/dp[0][S],这个是好朋友送了第一个礼物的概率，那么现在对于第一种礼物他有digit【0】个我们需要有1/digit[0]的概率去得到这个礼物，以此类推去得到第1个第2个等等。。。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string.h>
using namespace std;
const int maxn=;
const int MAXM=****+;
int gift[maxn],n;
double dp[][MAXM],to[][maxn];
bool vis[][MAXM];
int A[maxn];
double dfs(int loc, int G){
    if(vis[loc][G]) return dp[loc][G];
    int K=G;
    for(int i=; i>=; --i)
    {
         A[i] = K% ; K /=  ;
    }
    double ans=;
    for(int i=; i<; ++i)
    if( A[i] ){
           A[ i ]--;
           K = ;
           for( int j =  ; j< ; ++ j )
             K = K *  + A[ j ] ;
          ans+=dfs( loc +  , K ) * to[ loc ][ i ] ;
          A[i]++;
    }
    vis[loc][G]=true;
    return dp[loc][G]=ans;
}
int main()
{
   int cas;
   scanf("%d",&cas);
   while( cas -- ){
       scanf("%d",&n);
       for(int i=; i<; ++i)
        scanf("%d",&gift[i]);
       for(int i=; i<n; ++i)
        for(int j=; j<; ++ j)
           scanf("%lf",&to[i][j]);
       memset(dp,,sizeof(dp));
       memset(vis,false,sizeof(vis));
      vis[n][]=;
      dp[n][]=;
      int k=;
      for(int i=;i<; ++i ){
           k=k*+gift[i];
      }
      double res=dfs(,k);
      int  ans=;
      double pans=-;
      for(int i=; i<; ++i)if(gift[i]){
          gift[i]--;
          k=;
          for(int j=; j<; ++j)
            k=k*+gift[j];
            gift[i]++;
          double a=dp[][k]*to[][i]/gift[i]/res;
          if(a>pans){
             ans=i+; pans=a;
          }
      }
      printf("%d %.3lf\n",ans,pans);
   }
    return ;
}