用 Python 编写程序,在删除 K 个元素后查找最小振幅

pythonserver side programmingprogramming更新于 2025/12/24 16:44:17

假设我们有一个名为 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

相关文章


有用资源