Pythonで特定の配列式が最大化される値を見つけるプログラム

numsとvaluesという2つの配列があり、どちらにも整数が含まれていて、numsの値が厳密に増加していて、長さも同じであるとします。次のように、インデックスi、jのペアのvの値を見つける必要があります。v=values [i] + values [j] + nums [j]--nums[i]を最大化するi≤j。

したがって、入力が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