POJ3071 Football（概率DP）

POJ3071 Football

题干：

$2^n$个队伍打$n$轮淘汰赛，每轮都是相邻两个队伍打，决出胜者参与下一轮。$p[i][j]$表示队伍$i$打赢队伍$j$的概率。问比赛打完后，获胜概率最大的队伍是哪个。

解：

设$dp[i][j]$表示比赛$j$轮，第$i$号队伍获胜的概率。则：

$dp[i][j]=dp[i][j-1]\ast ({\sum }_{k\in opposite}p[i][k]\ast dp[k][j-1])$

$dp[i][0]=1$

问题是，在第$j$轮，$i$的对手$k$都有谁？

不妨来举例写一下（涉及二进制位运算，下标从0开始为好）：

设要打$3$轮淘汰赛，则会有$2^3=8$支队伍，给队伍从$0$开始标号，

第$1$轮是$(0,1),(2,3),(4,5),(6,7)$；

第$2$轮是$(0,1,2,3),(4,5,6,7)$，其中$(0,1),(2,3),(4,5),(6,7)$括号内的互相不能打比赛；

第$3$轮是$(0,1,2,3,4,5,6,7)$，其中$(0,1,2,3),(4,5,6,7)$括号内的互相不能打比赛。

第$1$轮比赛涉及的队伍范围内是$2^1=2$，第$2$轮涉及的队伍范围是$2^2=4$，第$3$轮涉及的队伍范围是$2^3=8$。队伍编号除以队伍范围可得：

第$1$轮是$(0,1),(2,3),(4,5),(6,7)$；

$2^1=2$ 【0】 【1】【2】 【3】

第$2$轮是$(0,1,2,3),(4,5,6,7)$，其中$(0,1),(2,3),(4,5),(6,7)$括号内的互相不能打比赛；

$2^2=4$ 【0】 【1】

第$3$轮是$(0,1,2,3,4,5,6,7)$，其中$(0,1,2,3),(4,5,6,7)$括号内的互相不能打比赛。

$2^3=8$ 【0】

发现第$j$轮比赛的队伍编号除以$2^j$，得到的数相同说明该轮可进行比赛；队伍编号除以$2^{j-1}$，得到的数相同说明上轮作为对手已经比赛过了。

所以得到第$j$轮比赛$k$是否作为$i$对手的判断条件（枚举队伍编号$k$判断是否是对手即可）：

if(((k>>j)==(i>>j)) && ((k>>(j-1))!=(i>>(j-1))))

for(int j = 1; j<= n; j++)
	for(int i = 0; i < m; i ++)
		for(int k = 0; k < m; k ++)
            if(((k>>j)==(i>>j))&&((k>>(j-1))!=(i>>(j-1)))){
                 dp[i][j] += (double)dp[i][j-1]*p[i][k]*dp[k][j-1];
            }

化简版的是直接列出$k$的所有取值。首先，对于第$j$轮，比赛队伍集合的长度是$2^j-2^{j-1}$，然后看一个集合的编号起始值是多少。

可知对于第$j$轮的第$i$号队伍，所在的比赛队伍集合的起始编号是$[\frac{i}{2^j}]\ast 2^j$，但是还要排除掉不能打的队伍。考虑上一轮的情况，如果编号$i$除以$2^{j-1}$是偶数，说明在第$j$轮小集合合并后，会跟新集合后半部分的队伍打，所以起始编号应该是原来的值$+2^{j-1}$；如果是奇数，说明在第$j$轮小集合合并后，会跟新集合前半部分的队伍打，起始编号即为$[\frac{i}{2^j}]\ast 2^j$。

所以：

st = ((i>>j)<<j) + (((i>>(j-1)) & 1)?0:(1<<(j-1)));
len = (1<<j)-(1<<(j-1));
ed = st - 1 + len;

for(int j = 1; j<= n; j++)
   for(int i = 0; i < m; i ++){
       int st = ((i>>j)<<j) + (((i>>(j-1)) & 1)?0:(1<<(j-1))), len = (1<<j)-(1<<(j-1));
       for(int k = st; k <= st-1+len; k++)
           dp[i][j] += (double)dp[i][j-1]*p[i][k]*dp[k][j-1];
    }