leetcode_486. 预测赢家 python3
目录
一、题目内容
给定一个表示分数的非负整数数组。 玩家 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 仍为赢家。
二、解题思路
类似博弈树的思想,这里我们可以计算每次当前玩家选择最左端或者最右端与之前的另一个玩家得到的分数总和的差值。
每次递归都会这样计算一遍,得到的是当前玩家(玩家1或玩家2)选择最左端和最右端后的差值的大者。
三、代码
-
class Solution:
-
def PredictTheWinner(self, nums):
-
-
def iter_max_min(left, right):
-
if left == right:
-
return nums[left]
-
a = nums[left] - iter_max_min(left + 1, right)
-
b = nums[right] - iter_max_min(left, right - 1)
-
return max(a, b)
-
return iter_max_min(0, len(nums) - 1) >= 0
-
-
-
if __name__ == '__main__':
-
test = [1, 5, 2]
-
s = Solution()
-
ans = s.PredictTheWinner(test)
-
print(ans)
文章来源: nickhuang1996.blog.csdn.net,作者:悲恋花丶无心之人,版权归原作者所有,如需转载,请联系作者。
原文链接:nickhuang1996.blog.csdn.net/article/details/108342692
- 点赞
- 收藏
- 关注作者
评论(0)