在 Python 中编写删除 K 个元素后找到最小振幅的程序
假设我们有一个名为 nums 的数字列表,还有一个值 k。如果我们从 nums 中删除 k 个元素,则查找 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]
- 如果 nums[i + p – 1] – nums[i] < m,则
- 返回 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