题目:给定一个表示分数的非负整数数组。 玩家1从数组任意一端拿取一个分数,随后玩家2继续从剩余数组任意一端拿取分数,然后玩家1拿,……。每次一个玩家只能拿取一个分数,分数被拿取之后不再可取。直到没有剩余分数可取时游戏结束。最终获得分数总和最多的玩家获胜。
给定一个表示分数的数组,预测玩家1是否会成为赢家。你可以假设每个玩家的玩法都会使他的分数最大化。
输入: [1, 5, 2] 输出: False
解释: 一开始,玩家1可以从1和2中进行选择。
如果他选择2(或者1),那么玩家2可以从1(或者2)和5中进行选择。
如果玩家2选择了5,那么玩家1则只剩下1(或者2)可选。
所以,玩家1的最终分数为 1 + 2 = 3,而玩家2为 5。因此,玩家1永远不会成为赢家,返回 False。
解决方案:
我们同样使用动态规划来解决这个问题。用 dp(i, j) 表示当剩下的数为 nums[i .. j] 时,
当前操作的选手(注意,不一定是先手)与另一位选手最多的分数差。
当前操作的选手可以选择 nums[i] 并留下 nums[i+1 .. j],或选择 nums[j] 并留下 nums[i .. j-1],
因此状态转移方程为:
dp(i, j) = max(nums[i] - dp(i+1, j), nums[j] - dp(i, j-1))
dp(i, i) = nums[i]
如果 dp(0, n - 1) >= 0,那么先手必胜。
注意:因为i依赖i+1,j依赖j-1 所以i倒序遍历,j正序遍历。
又因为dp方程有i+1,所以i从len-2开始取,同样方程中有j-1,j从i+1开始取
public boolean PredictTheWinner(int[] nums) {
int len = nums.length;
if (len <= 2) return true;
int[][] dp = new int[len][len];
for (int i = len - 2; i >= 0; --i) {// 有i+1,所有i取len-2
for (int j = i + 1; j < len; ++j) {// 有j-1,所以j取i+1
dp[i][j] = Math.max(
nums[i] - dp[i+1][j],
nums[j] - dp[i][j - 1]
);
}
}
return dp[0][len - 1] >= 0;
}
对于上面dp方程,我们发现 dp(i, j) 只和 dp(i+1, j) 和 dp(i, j-1) 有关
,即在计算第 i 行的 dp 值时,只有该行与第 i + 1 行是有用的。因此我们可以将空间列优化至一维。
public boolean PredictTheWinner(int[] nums) {
int len = nums.length;
if (len <= 2) return true;
int[] dp = new int[len];
for (int i = len - 2; i >= 0; --i) {// 有i+1,所有i取len-2
for (int j = i + 1; j < len; ++j) {// 有j-1,所以j取i+1
dp[j] = Math.max(nums[i] - dp[j], nums[j] - dp[j - 1]);
}
}
return dp[len - 1] >= 0;
}