zoj-3735 Josephina and RPG[概率dp]

/*就是告诉你C(m,3)个队伍相互之间的胜率，
然后要你依次对战n个AI队伍，首先任选一种队伍，
然后战胜一个AI后可以选择替换成AI的队伍，也可以不换，
问你最后最大的胜率是多少。*/

状态跟对战队伍的顺序有关，跟玩家使用的对战队伍有关，对战后有替换和不替换两种转变。

那么设状态方程为dp[i][j] i表示对战第i个队伍,j表示战胜后使用第j个队伍最大的胜率。

对 dp[i][j] 向dp[i+1][]转移的考虑

for(j:0....n)//列举队伍

不替换时,保持j的继续使用，dp[i+1][j]就是对dp[i][j]的累积

替换时，解决num[i+1]依然是j，但是以后累积的队伍是num[i+1]，所以j累积的状态应转移到dp[i+1][num[i+1]]

			//在打败i+1队后不换队伍
			dp[i+1][j]=max(dp[i][j]*g[j][num[i+1]],dp[i+1][j]);
			//在打败i+1队后换队伍 
			dp[i+1][num[i+1]]=max(dp[i][j]*g[j][num[i+1]],dp[i+1][num[i+1]]);

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <algorithm>
using namespace std;
const int maxm=128;
const int maxn=10008;
const double eps=1e-6;
double dp[maxn][maxm];//战胜第i个队伍后用的队伍是j 
double g[maxm][maxm];//i对j的胜率 
int num[maxn];
int fun(int m)
{
	int res=1;
	for(int i=0,j=m;i<3;++i,--j)
		res*=j;
	res/=6;
	return res;
}
void DP(int n,int t)
{
	memset(dp,0,sizeof(dp));
	for(int i=0;i<n;++i)
		dp[0][i]=1;
	for(int i=0;i<t;++i)
	{
		for(int j=0;j<n;++j)
		{
			//在打败i+1队后不换队伍
			dp[i+1][j]=max(dp[i][j]*g[j][num[i+1]],dp[i+1][j]);
			//在打败i+1队后换队伍 
			dp[i+1][num[i+1]]=max(dp[i][j]*g[j][num[i+1]],dp[i+1][num[i+1]]);
		}
	}
	double ans=0;
	for(int i=0;i<n;++i)
		ans=max(dp[t][i],ans);
	printf("%.6lf\n",ans);
} 
int main()
{
	int m;
	while(~scanf("%d",&m))
	{
		int n=fun(m); 
		for(int i=0;i<n;++i)
			for(int j=0;j<n;++j)
				scanf("%lf",&g[i][j]);
		int t;
		scanf("%d",&t);
		for(int i=1;i<=t;++i)
			scanf("%d",num+i);
		DP(n,t);
	}
	return 0;
}