在 Python 中编写程序检查是否可以通过得到最高分数来赢得糖果游戏

假设有两个玩家在玩一个游戏。在游戏中有许多糖果被放置在一条线上，给出一个列表，叫做 nums，表示每个糖果的点数。每个人轮流从排列在最前面的 1、2 或 3 个糖果中挑选一些糖果，从列表中删除，然后将他们的点数总和加到他们的分数中。所有糖果被删除时游戏结束，得分更高的人将成为获胜者。我们需要检查第一个人是否能赢得这场比赛。

所以，如果输入是 nums = [1，1，2，3，50]，那么输出将是 True，因为第一人可以拿走 1 个糖果，然后另一个玩家必须拿走 1、2 或 3 个糖果。不管怎样，第一人都可以拿到价值为 50 的糖果。

为了解决这个问题，我们将遵循以下步骤 –

n := nums 的大小
table := 具有三个 0 的数组。
对于 i 在范围 n – 1 到 0，按步长为 1 递减，执行以下操作 –
- profit := -inf
- sum_val := 0
- 对于 j 在范围 i 到 i + 3 和 n 的最小值之间，进行以下操作 –
  - sum_val := sum_val + nums[j]
  - profit := profit 和 (sum_val – table[j – i]) 中的最大值。
- 设置table:= 具有三个值 [profit, table[0], table[1]] 的列表。
当 table[0]> 0 时返回 true，否则返回 false

让我们看下面的实现，以更好的理解 –

例子

import math
class Solution:
  def solve(self, nums):
    n = len(nums)
    table = [0, 0, 0]
    for i in range(n - 1, -1, -1):
      profit = -math.inf
      sum_val = 0
      for j in range(i, min(i + 3, n)):
        sum_val += nums[j]
        profit = max(profit, sum_val - table[j - i])
      table[:] = [profit, table[0], table[1]]
    return table[0] > 0
ob = Solution()
nums = [1, 1, 2, 3, 50]
print(ob.solve(nums))