在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