您的位置:首页 > 文旅 > 旅游 > 【LeetCode】每日一题:数组中的第K大的元素

【LeetCode】每日一题:数组中的第K大的元素

2024/10/6 20:26:01 来源:https://blog.csdn.net/qq_44459787/article/details/139997008  浏览:    关键词:【LeetCode】每日一题:数组中的第K大的元素

给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。
请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。

解题思路

第一种是快排,快排逻辑是以一个元素作为哨兵,通过头尾指针逼近和交换元素的方法找到该哨兵的位置,此题中额外使用k进行剪枝。

第二种思路是使用堆heapify,这种方式会默认生成一个大根堆,可以通过“ListNode.__lt__ = lambda a, b: a.val < b.val # 让堆可以比较节点大小”,然后直接使用heappop返回当前最小值。

AC代码

class Solution:def findKthLargest(self, nums: List[int], k: int) -> int:# def quicksort(nums, l, r, k):#     if l == r:#         return nums[k]#     i, j, key = l, r, nums[l]#     while i < j:#         while nums[i] < key: i += 1#         while nums[j] > key: j -= 1#         if i < j:#             nums[i], nums[j] = nums[j], nums[i]#     return quicksort(nums, l, j, k) if k <= j else quicksort(nums, i+1, r, k)# return quicksort(nums, 0, len(nums) - 1, k)heapify(nums)temp = 0for _ in range(len(nums) - k + 1):temp = heappop(nums)return temp    

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com