Python | 找到大于 K 的最小元素

本文介绍如何在 Python 中找到大于给定数 K 的最小元素。

方法一：使用迭代

def find_min_greater_than_K(nums, K):
    min_greater_than_K = float('inf')
    for num in nums:
        if num > K and num < min_greater_than_K:
            min_greater_than_K = num
    return min_greater_than_K

使用迭代的方法比较简单，我们依次遍历列表中的每个元素，如果发现元素大于 K 且小于当前最小值 min_greater_than_K，则更新最小值。最后返回最小值即可。

• nums：输入的列表。
• K：给定的数。

方法二：使用列表推导式

def find_min_greater_than_K(nums, K):
    return min([num for num in nums if num > K])

我们可以使用列表推导式来一行代码实现上述功能，关键在于要保证列表非空。如果列表中没有大于 K 的元素，我们使用函数 min() 计算会返回整型或浮点型最大值。

示例：

>>> nums = [2, 5, 7, 8, 10, 11]
>>> K = 6
>>> find_min_greater_than_K(nums, K)
7

以上示例说明，给定列表 nums 和数 K，我们成功找到了大于 K 的最小元素 7。

以上是本文的全部内容，我们介绍了两种方法来找到大于给定数 K 的最小元素。使用哪种方法取决于您的喜好和代码实现的复杂度。