HDU 4800 Josephina and RPG【概率DP】

题目链接

题意：有C(m,3)个队伍，已知第i个队伍在面对第j个队伍时的胜率。现在要按顺序打败n支队伍，可以任选一支队伍开始，每次打赢后可以选择不换队伍或者换成刚才打赢的那支队伍，求最后能打败n支队伍的概率。

很明显的概率DP，dp[i][j]表示击败前i支队伍后用的队伍是j。那么击败队伍i后只有两种可能，换或者不换。如果不换，用的还是队伍j，dp[i][j] = max(dp[i][j], dp[i - 1][j] * p[j][w[i]])。如果换了，这时用的是队伍w[i]，dp[i][w[i]] = max(dp[i][w[i]], dp[i - 1][j] * p[j][w[i]])。

WA了很久，错在初始化上，应该是击败前0支队伍后用任何一支队伍的概率都是1，即dp[0][0~n-1]=1。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
double p[200][200];
double dp[10000 + 10][200];
int w[10000 + 10];
int nn, n, m;

int C(int M)
{
	return M * (M - 1) * (M - 2) / 6;
}

int main()
{
	while (scanf("%d", &nn) != EOF)
	{
		n = C(nn);
		for (int i = 0; i < n; i++)
			for (int j = 0; j < n; j++)
				scanf("%lf", &p[i][j]);
		int m;
		scanf("%d", &m);
		for (int i = 1; i <= m; i++)   scanf("%d", &w[i]);
		memset(dp, 0, sizeof(dp));
		for (int i = 0; i < n; i++)	dp[0][i] = 1;
		for (int i = 1; i <= m; i++)
		{
			for (int j = 0; j < n; j++)
			{
				dp[i][j] = max(dp[i][j], dp[i - 1][j] * p[j][w[i]]);
				dp[i][w[i]] = max(dp[i][w[i]], dp[i - 1][j] * p[j][w[i]]);
			}
		}
		double ans = 0;
		for (int i = 0; i < n; i++)	ans = max(ans, dp[m][i]);
		printf("%.6f\n", ans);
	}
	return 0;
}