题目
亚历克斯和李继续他们的石子游戏。许多堆石子 排成一行,每堆都有正整数颗石子 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;
}
}
图片可帮助理解代码。