题目链接
http://poj.org/problem?id=3071
题解
题目大意:2^n个人比赛,每相邻的比一下决出胜者再相邻比,每两个人比拼有概率输赢,问最后谁获胜的概率最大。
明显比赛的关系是一棵二叉树,预处理出每个人j在第i轮获胜后的位置pos[i][j]。
设f[i][j]为第i轮j胜出的概率,枚举j在第i轮的对手k,则f[i][j] += f[i-1][k] * f[i-1][j] * p[j][k],直接转移即可。(转移的条件是j,k在第i轮获胜后去的位置相同且第i-1轮不同,即pos[i][j] == pos[i][k] && pos[i-1][j] != pos[i-1][k])
如何才能WA呢?多组数据不清空数组即可。
时间复杂度O(n*4^n*T)。
代码
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <cstdlib>
#define maxn 200
using namespace std;
int n, m;
double f[maxn][maxn];
double p[maxn][maxn];
int pos[maxn][maxn];
int main(){
while(~ scanf("%d", &n) && (~ n)){
m = 1;
for(int i = 1; i <= n; i++) m <<= 1;
for(int i = 1; i <= m; i++)
for(int j = 1; j <= m; j++)
scanf("%lf", &p[i][j]);
for(int i = 1; i <= m; i++){
pos[0][i] = i;
for(int j = 1; j <= n; j++)
pos[j][i] = (pos[j-1][i] + 1) >> 1;
}
for(int i = 1; i <= m; i++)
f[0][i] = 1.0;
for(int i = 1; i <= n; i++){
for(int j = 1; j <= m; j++){
f[i][j] = 0.0;
for(int k = 1; k <= m; k++){
if(pos[i][j] == pos[i][k] && pos[i-1][j] != pos[i-1][k])
f[i][j] += f[i-1][k] * f[i-1][j] * p[j][k];
}
}
}
double Max = f[n][1];
int ans = 1;
for(int i = 2; i <= m; i++){
if(Max < f[n][i]){
Max = f[n][i];
ans = i;
}
}
printf("%d\n", ans);
}
return 0;
}