一.题目描述
给定一个表示分数的非负整数数组。 玩家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可以成为赢家。
注意:
1 <= 给定的数组长度 <= 20.
数组里所有分数都为非负数且不会大于10000000。
如果最终两个玩家的分数相等,那么玩家1仍为赢家。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/predict-the-winner
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二.思路分析
首先考虑暴力解法,先手每一步都有两个选择,取左边或者取右边,每一步的取值都会对最终的结果产生影响,这样的话总的解的空间为O(2^n),如果使用暴力解法的话时间复杂度就为O(2^n),n为输入数组的长度
既然这样不行,那么我们换一种思路,使用动态规划。
动态规划的思想是自底向上,上面说的暴力解法是自顶向下,于是我们可以从这个算法的结尾开始考虑:只剩两个数组,比如[3,5],那么很显然先手取5会赢。然后如果输入的是[7,3,5],那么选择7还是5取决于后手在[3,5]或者[7,3]中作为先手的选择
每个步骤的状态我们可以定义为先手能赢到的所出来的分数
因此递推关系式为:
DP[i][j]=max{nums[i]-DP[i+1][j],nums[j]-DP[i][j-1]}
DP[i][j]表示在区间nums[i to j]中先手能够多赢的分数
也就是说分为先手选择nums[i](左边)还是nums[j](右边)两种情况
当选择nums[i]时,后手在选择nums[i+1 to j]时变成了先手,DP[i+1][j]表示他在这个区间作为先手选择能多赢的分数
原来的先手选择nums[i]时多赢的分数变成了nums[i]-DP[i+1][j],
选择nums[j]时也是同样的意思
这是一个在区间上进行动态规划的问题,就是当前区间的结果依赖前面区间的结果
同样的做游戏是否能赢的题目还有:
https://leetcode-cn.com/problems/can-i-win/
猜数字大小二
https://leetcode-cn.com/problems/guess-number-higher-or-lower-ii/
代码如下:
class Solution:
def PredictTheWinner(self, nums) -> bool:
Len=len(nums)
DP=[[0 for i in range(Len)] for j in range(Len)]
for i in range(Len):
DP[i][i]=nums[i]
for i in range(2,Len+1):
for start in range(0,Len-i+1):
DP[start][start+i-1]=max(nums[start]-DP[start+1][start+i-1],nums[start+i-1]-DP[start][start+i-2])
return DP[0][Len-1]>=0