用 Python 编写程序，求出子序列的最大和，其中两个值的差与它们的位置差相同

pythonserver side programmingprogramming更新于 2026/2/4 2:52:17

假设我们有一个名为 nums 的数字列表，我们选择一个严格递增的值的子序列，其中每两个数字的差与它们两个索引的差相同。所以我们必须找到这样一个子序列的最大和。

因此，如果输入为 nums = [6, 7, 9, 9, 8, 5]，则输出将为 22，因为我们选择了索引为 [0, 1, 3] 的子序列 [6, 7, 9]。每个连续数字之间的差异是 [1, 2]，与它们的索引差异相同。

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

d := 一个空映射
对于 nums 中的每个索引 i 和值 x，执行
- d[x − i] := d[x − i] + x
返回 d 中所有值的最大值

让我们看看下面的实现以便更好地理解 −

示例

class Solution:
   def solve(self, nums):
      from collections import defaultdict
      d = defaultdict(int)
      for i, x in enumerate(nums):
         d[x − i] += x
      return max(d.values())

ob1 = Solution()
nums = [6, 7, 9, 9, 8, 5]
print(ob1.solve(nums))