【Lintcode】325. Football

题目地址：

https://www.lintcode.com/problem/325/description

有$16$支球队，编号$0\sim 15$，每一轮先将所有球队按照编号从小到大排序，然后第$2k+1,2k+2$支球队比赛，胜者进入下一轮，从而一共会比$4$轮。给定一个$16\times 16$的方阵$p$，$p[i][j]$是队伍$i$战胜队伍$j$的概率。问哪支球队得到最后的冠军的概率最高。

设$f[k][i]$是经过$k$轮比赛之后队伍$i$仍然存活的概率，那么就是要求${\max }_{i}f[4][i]$。对于队伍$i$，如果他能遇到的队伍的编号是集合$S$，那么有：$$f[k][i]=\sum _{j\in S}f[k-1][i]\times f[k-1][j]\times p[i][j]$$接下来就需要确定$i$能遇到哪些队伍。可以将整个系列赛看成一棵二叉树，两个节点父亲相同的时候就会比赛，胜利的队伍爬上去作为父亲节点。在第一轮的时候，$i\wedge 1=j$的时候两个队伍会比赛；而每个队伍进入下一轮之后，下标就变成了$i/2$。所以第$k$轮两个队伍可能遇到当且仅当$i/2^{k-1}\wedge 1=j/2^{k-1}$。代码如下：

import java.util.Arrays;

public class Solution {
    /**
     * @param probability: the winning probability of 16 teams in pairs
     * @return: return the index of the winner with the highest probability of winning
     */
    public int findWinner(double[][] probability) {
        // write your code here
        double[][] f = new double[5][16];
        Arrays.fill(f[0], 1.0);
        
        for (int k = 1; k <= 4; k++) {
            for (int i = 0; i < 16; i++) {
                for (int j = 0; j < 16; j++) {
                    if ((i >> k - 1 ^ 1) == (j >> k - 1)) {
                        f[k][i] += f[k - 1][i] * f[k - 1][j] * probability[i][j];
                    }
                }
            }
        }
        
        // 求获胜概率最大的队伍编号
        int res = 0;
        for (int i = 0; i < 16; i++) {
            if (f[4][i] > f[4][res]) {
                res = i;
            }
        }
        
        return res;
    }
}

时空复杂度$O(1)$。