Leetcode 486:预测赢家

给定一个表示分数的非负整数数组。 玩家1从数组任意一端拿取一个分数,随后玩家2继续从剩余数组任意一端拿取分数,然后玩家1拿,……。每次一个玩家只能拿取一个分数,分数被拿取之后不再可取。直到没有剩余分数可取时游戏结束。最终获得分数总和最多的玩家获胜。

给定一个表示分数的数组,预测玩家1是否会成为赢家。你可以假设每个玩家的玩法都会使他的分数最大化。

示例 1:
输入: [1, 5, 2]
输出: False
解释: 一开始,玩家1可以从12中进行选择。
如果他选择2(或者1),那么玩家2可以从1(或者2)和5中进行选择。如果玩家2选择了5,那么玩家1则只剩下1(或者2)可选。
所以,玩家1的最终分数为 1 + 2 = 3,而玩家25。
因此,玩家1永远不会成为赢家,返回 False。

示例 2:
输入: [1, 5, 233, 7]
输出: True
解释: 玩家1一开始选择1。然后玩家2必须从57中进行选择。无论玩家2选择了哪个,玩家1都可以选择233。
最终,玩家1234分)比玩家212分)获得更多的分数,所以返回 True,表示玩家1可以成为赢家。

注意:
1 <= 给定的数组长度 <= 20.
数组里所有分数都为非负数且不会大于10000000。
如果最终两个玩家的分数相等,那么玩家1仍为赢家。
解题思路

这个问题非常简单,我们需要考虑的就是从左拿取一个数还是从右拿取一个数?这是由什么决定的?最好的情况当然是p1取的数很大,而p2取的数很小,也就是二者取的数的差值越大越好。那么我们可以定义函数f(l, r)表示我们每次从左还是右取数,那么函数的返回值
f(l,r)=max(nums[l]−f(l+1,r),nums[r]−f(l,r−1))
直接思考边界问题,也就是当l==r的时候,此时显然f(l,r)=nums[l]

class Solution:
    def PredictTheWinner(self, nums: List[int]) -> bool:
        def right_or_left(r,l):
            if r==l:
                return nums[r]
            return max(nums[r]-right_or_left(r+1,l),nums[l]-right_or_left(r,l-1))
        aa=right_or_left(0,len(nums)-1)
        if aa >= 0:
            return True
        else:
            return False

从运行时间我们可以看到这一步的优化操作有多么重要。我们也可以手动实现一个记忆化的过程。

class Solution:
    def PredictTheWinner(self, nums: 'List[int]') -> 'bool':
        n = len(nums)
        mem = [[0]*n for _ in range(n)]
        
        def selectNumber(l, r):
            if l == r: 
                return nums[l]
            if mem[l][r] > 0:
                return mem[l][r]
            mem[l][r] = max(nums[l] - selectNumber(l+1, r), nums[r] - selectNumber(l, r-1))
            return mem[l][r]
        return True if selectNumber(0, n-1) >= 0 else False

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值