在Python中查找使给定数组表达式最大化的值

假设我们有两个名为nums和values的数组，两个数组都包含整数，nums的值是严格递增的，它们的长度也相同。我们必须找到对于一对索引i，j，使以下方程式最大化的v的值：i ≤ j，其中v = values [i] + values [j] + nums [j] – nums [i]。

因此，如果输入是nums =[1,2,7] values=[-4,6,5]，则输出将为16。如果我们选择i = 1和j = 2，我们得到6 + 5 + 7-2 = 16。

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

• ans1：= -inf，ans2：= -inf

• 对于区间在0到nums大小-1的i，执行以下操作

  • ans1：= ans1和（values [i] – nums [i]）的最大值

  • ans2：= ans2和（values [i] + nums [i]）的最大值

• 返回ans1 + ans2

示例

让我们看一下以下实现，以获取更好的理解

from math import inf
def solve(nums, values):
   ans1 = -inf
   ans2 = -inf
   for i in range(len(nums)):
      ans1 = max(ans1, (values[i] - nums[i]))
      ans2 = max(ans2, (values[i] + nums[i]))
   return ans1 + ans2

nums = [1, 2, 7]
values = [-4, 6, 5]
print(solve(nums, values))

输入

[1, 2, 7]，[-4，6，5]

输出

16