在 Python 中找出操作后最大子数组的总和的程序

假设,我们得到一个包含整数的数组。我们可以执行一个操作,将 array[i] 的值替换为其平方值;或数组[i] * 数组[i]。只允许进行一次此类操作,并且我们必须在操作后返回最大可能子数组的总和。子数组不能为空。

因此,如果输入类似于 array = [4, 1, -2, -1],那么输出将是 17。

如果我们用其平方值替换 array[0] 中的值,则数组变为 [16, 1, -2, -1]。由此可能产生的最大子数组是 [16, 1],它的最大和值为 16 + 1 = 17。

示例

让我们看看以下实现以获得更好的理解 -

def solve(array):

   dp1 = [float('-inf')]

   dp2 = [float('-inf')]

   for num in array:

      dp1.append(max(dp1[-1] + num, num))

      dp2.append(max(dp1[-2] + num**2, num**2, dp2[-1]+num))

   return max(dp2)

print(solve([4, 1, -2, -1]))

输入

[4, 1, -2, -1]
输出结果
17

以上是 在 Python 中找出操作后最大子数组的总和的程序 的全部内容, 来源链接: utcz.com/z/358579.html

回到顶部