用 Python 编写程序,查找子数组中最小增量数以形成目标数组

pythonserver side programmingprogramming更新于 2025/12/1 1:48:17

假设我们有一个名为 target 的数组,其值为正数。现在考虑一个初始大小相同且全为零的数组。如果我们执行此操作,我们必须找到从初始生成目标数组所需的最少操作数:(从初始中选择任何子数组 并将每个值增加一。)

因此,如果输入类似于目标 = [2,3,4,3,2],则输出将为 4,因为初始数组为 [0,0,0,0,0],首先从索引 0 到 4 选择子数组并将其增加 1,因此数组将为 [1,1,1,1,1],然后再次从索引 0 到 4 选择以使其为 [2,2,2,2,2],然后从索引 1 到 3 选择元素并增加,因此数组将为 [2,3,3,3,2],最后选择索引 2 并使数组 [2,3,4,3,2] 与目标相同。

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

  • prev_num := 0

  • steps := 0

  • 对于目标中的每个 val,执行

    • steps := steps + val - prev_num 如果 val > prev_num 否则为 0

    • prev_num := val

  • 返回 steps

示例

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

def solve(target):
   prev_num = 0
   steps = 0
   for val in target:
      steps += val-prev_num if val > prev_num else 0
      prev_num = val
   return steps

target = [2,3,4,3,2]
print(solve(target))

输入

[2,3,4,3,2]

输出

4

相关文章


有用资源