Football (概率dp求概率)

Football

POJ - 3071

大致题意:

2*n只球队,按顺序依次进行比赛,一轮下来,胜利的队伍再按顺序依次进行比赛,给出i队战胜j队的概率

求一个最大概率赢的队伍

---

解题思路:

概率dp求概率一般是正推

状态表示:f[i][j] 表示进行到第i回合,第j只队伍获胜的概率

初始化 f[0][j]=0, 答案取max(f[n][j])

分析: f[i][j]表示第i回合j队获胜,那么他要在上一回合获胜,即f[i-1][j],此外,他的对手k也要在上一回合获胜,即f[i-1][k], j队战胜k队的概率是p[j][k] ,所以 f[i][j]=f[i-1][j]*f[i-1][k]*f[j][k] ,转移方程就出来了,还有一个问题,就是j队和k队要满足两个队之间是第一次比赛,这里就会多一个转移的条件,用二进制表示比赛进程的话,当j>>(i-1) 等于 (k>>(i-1))^1时,表明两个队之间是第一次比赛

转移方程:f[i][j]=f[i-1][j]*f[i-1][k]*f[j][k]

补充:因为是二进制表示比赛进程,所以下标是从0开始

---

AC代码:

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
#define debug(a) cout << #a << " = " << a << endl;
using namespace std;
typedef long long ll;
const int N = 210;
int n, m;
double f[N][N];   //表示第i回合第j队赢得概率
double p[N][N];   //表示i队打败j队的概率
int main(void)
{
	while (~scanf("%d", &n) && n != -1) {
		m = 1 << n;
		memset(f, 0, sizeof f);
		for (int i = 0; i < m; ++i)f[0][i] = 1;

		for (int i = 0; i < m; ++i)
			for (int j = 0; j < m; ++j)
				scanf("%lf", &p[i][j]);

		for (int i = 1; i <= n; ++i)
			for (int j = 0; j < m; ++j)
				for (int k = 0; k < m; ++k)
					if ((j >> (i - 1)) == ((k >> (i - 1)) ^ 1)) {   //条件
						f[i][j] += f[i - 1][j] * p[j][k] * f[i - 1][k];
					}
		
		int res = 0;
		for (int i = 1; i < m; ++i)
			if (f[n][i] > f[n][res])
				res = i;
		printf("%d\n", res + 1);
	}
	return 0;
}

---