🚩传送门:https://leetcode-cn.com/problems/predict-the-winner/
题目
给定一个表示分数的非负整数数组。 玩家 1 从数组任意一端拿取一个分数,随后玩家 2 继续从剩余数组任意一端拿取分数,然后玩家 1 拿,…… 。每次一个玩家只能拿取一个分数,分数被拿取之后不再可取。直到没有剩余分数可取时游戏结束。最终获得分数 总和 最多的玩家获胜。
给定一个表示分数的数组,预测 玩家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。
示例 2:
输入:[1, 5, 233, 7] 输出:True 解释:
玩家1一开始选择1。然后玩家2必须从5和7中进行选择。无论玩家2选择了哪个,玩家1都可以选择233。最终,玩家1(234分)比玩家2(12分)获得更多的分数,所以返回True,表示玩家1可以成为赢家。
解题思路:递归
为了判断哪个玩家可以获胜,需要计算一个总分,即先手得分与后手得分 之差。当数组中的所有数字都被拿取时,如果总分大于或等于 0,则先手获胜,反之则后手获胜。
每次只从数组的任意一端拿取数字,因此利用双指针滑动可以保证指针间一定是连续的。假设数组当前剩下的部分为下标 到下标 ,其中 。
若 ,则只剩一个数字,当前玩家只能拿取这个数字。
若 ,则当前玩家可以选择 或 ,另一个玩家在数组剩下的部分选取
计算总分时,需要记录当前玩家是先手还是后手,判断当前玩家的得分应该记为正还是负。![[LeetCode]Dp486 预测赢家 [滚动数组] - 图1](/uploads/projects/mylearn@leetcode/33f626a40f7c330ad56f3588392be154.png)
复杂度分析
时间复杂度:,其中 为数组 的长度 。
空间复杂度:,其中 为数组 的长度。空间复杂度取决于递归使用的栈空间。
代码
🚩官方代码
class Solution {public boolean PredictTheWinner(int[] nums) {return total(nums, 0, nums.length - 1, 1) >= 0;}public int total(int[] nums, int start, int end, int turn) {if (start == end) {return nums[start] * turn;}int scoreStart = nums[start] * turn + total(nums, start + 1, end, -turn);int scoreEnd = nums[end] * turn + total(nums, start, end - 1, -turn);return Math.max(scoreStart * turn, scoreEnd * turn) * turn;}}
解题思路:动态规划
👉 定义 为在区间 里,当前玩家与另一个玩家的分差的最大值 ,当前玩家不一定是先手。
不难发现只有当 时,数组剩下的部分才有意义,因此当 时,。
- 当 时,只剩一个数,只能取此数,故对于 ,有
- 当 时,当前玩家可以选择 或 ,然后轮到另一个玩家在剩余数字中选取。
- 两种方案中,当前玩家会选择最优方案,使自己的分数最大化。因此可以推出状态转移方程。
- 最后判断 的值,如果大于或等于 ,则先手得分大于或等于后手得分,因此先手成为赢家,否则后手成为赢家。
状态转移方程:

图解
复杂度分析
时间复杂度:,其中 为数组 的长度 。需要计算每个子数组对应的 的值,所以共
有 个子数组。
空间复杂度:,其中 为数组 的长度。若使用一维滚动数组优化可以降至 。
代码
🚩官方代码
class Solution {public boolean PredictTheWinner(int[] nums) {int length = nums.length;int[][] dp = new int[length][length];for (int i = 0; i < length; i++) {dp[i][i] = nums[i];}for (int i = length - 2; i >= 0; i--) {for (int j = i + 1; j < length; j++) {//状态转移方程dp[i][j] = Math.max(nums[i] - dp[i + 1][j], nums[j] - dp[i][j - 1]);}}return dp[0][length - 1] >= 0;}}
滚动数组优化

代码中使用了二维数组 dp。分析状态转移方程可以看到,dp[i][j] 的值只和 dp[i+1][j] 与 dp[i][j−1] 有关,即在计算 dp 的第 i 行的值时,只需要使用到 dp 的第 i 行和第 i+1 行的值,因此可以使用滚动数组对空间进行优化。
class Solution {public boolean PredictTheWinner(int[] nums) {int length = nums.length;int[] dp = new int[length];//初始化dp数组for (int i = 0; i < length; i++) {dp[i] = nums[i];}//外循环次数for (int i = length - 2; i >= 0; i--) {//内循环for (int j = i + 1; j < length; j++) {//状态转移方程dp[j] = Math.max(nums[i] - dp[j], nums[j] - dp[j - 1]);}}return dp[length - 1] >= 0;}}

