Python程序：删除K个元素后找到最小振幅

假设我们有一个称为nums的数字列表，还有一个值k。如果我们从nums中删除k个元素，则找到（nums的最大值 - nums的最小值）的最小值。

因此，如果输入类似于nums = [4, 10, 3, 2, 8, 9] k = 3，则输出将为2，因为如果我们删除10、8和9，则最大值为4，最小值为2，因此差值为2。

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

• 对列表nums进行排序

• p := nums的大小 - k

• m := (nums的最后一个元素) - nums[0]

• 对于范围从0到nums的大小 - p的i，执行以下操作：

  • 如果nums[i + p - 1] - nums[i] < m，则

    • m := nums[i + p - 1] - nums[i]

• 返回m

示例

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

def solve(nums, k):
   nums = sorted(nums)
   p = len(nums) - k
   m = nums[-1] - nums[0]
   for i in range(0, len(nums) - p + 1):
      if nums[i + p - 1] - nums[i] < m:
         m = nums[i + p - 1] - nums[i]
   return m

nums = [10, 4, 3, 2, 9, 8]
k = 3
print(solve(nums, k))

输入

[10, 4, 3, 2, 9, 8], 3

输出

2

Arnab Chakraborty

更新于: 2021年10月12日

409 次浏览

开启你的 职业生涯

通过完成课程获得认证

开始学习