力扣 877. 石子游戏

题目

亚历克斯和李继续他们的石子游戏。许多堆石子 排成一行，每堆都有正整数颗石子 piles[i]。游戏以谁手中的石子最多来决出胜负。

亚历克斯和李轮流进行，亚历克斯先开始。最初，M = 1。

在每个玩家的回合中，该玩家可以拿走剩下的 前 X 堆的所有石子，其中 1 <= X <= 2M。然后，令 M = max(M, X)。

游戏一直持续到所有石子都被拿走。

假设亚历克斯和李都发挥出最佳水平，返回亚历克斯可以得到的最大数量的石头。

示例：

输入：piles = [2,7,9,4,4]
输出：10
解释：
如果亚历克斯在开始时拿走一堆石子，李拿走两堆，接着亚历克斯也拿走两堆。在这种情况下，亚历克斯可以拿到 2 + 4 + 4 = 10 颗石子。 
如果亚历克斯在开始时拿走两堆石子，那么李就可以拿走剩下全部三堆石子。在这种情况下，亚历克斯可以拿到 2 + 7 = 9 颗石子。
所以我们返回更大的 10。 

提示：

1 <= piles.length <= 100
1 <= piles[i] <= 10 ^ 4

解答

可以看出，这也是一个动态规划问题，详细的解答思路见代码注释部分。

本质上还是把一个问题拆分成多个子问题，并采用dp数组优化，减少重复计算。

class Solution {
    public boolean stoneGame(int[] piles) {
        int n = piles.length;
        // 初始化dp数组
        int[][][] dp = new int[n][n][2]; //dp[i][j][k]，i,j表示piles[i...j]起始和结束的位置，k=0/1表示先/后手拿的最多石子数
        // 填入base case
        for(int i = 0; i < n; i++){
            dp[i][i][0] = piles[i];  //只有一堆石子，先手必然为这堆石子数
            dp[i][i][1] = 0;    //后手为0
        }
        /*
        在piles[i...j]中，状态转移方程：
        i...j中先手选了左边，那么在i+1...j中就成了后手，left = piles[i] + dp[i + 1][j].sec
        i...j中先手选了右边，那么在i...j-1中就成了后手，right = piles[i] + dp[i][j - 1].sec

        dp[i][j].fir = Math.max(piles[i] + dp[i + 1][j].sec, dp[i][j - 1].sec + piles[j]);
        if 先手选择左边:
            dp[i][j].sec = dp[i + 1][j].fir;  i...j的后手值是范围在i+1...j中的先手值
        if 先手选择右边:
            dp[i][j].sec = dp[i][j - 1].fir;
        可以看出dp[i][j]的值与dp[i + 1][j]、dp[i][j - 1]有关
        因此选择从下往上，从左往右遍历
        */
        for(int i = n - 2; i >= 0; i--){  //从倒数第二行开始
            for(int j = i + 1; j < n; j++){  //从对角线右边第一个开始
                int left = piles[i] + dp[i + 1][j][1];   //先手选左边的最大石子数
                int right = piles[j] + dp[i][j - 1][1];  //先手选右边的最大石子数
                if(left > right){  //先手选择左边
                    dp[i][j][0] = left;
                    dp[i][j][1] = dp[i + 1][j][0];
                }                
                else{  //先手选择右边
                    dp[i][j][0] = right;
                    dp[i][j][1] = dp[i][j - 1][0];
                }
            }
        }
        return dp[0][n - 1][0] - dp[0][n - 1][1] > 0;
    }
}

图片可帮助理解代码。