215. Kth Largest Element in an Array

最新推荐文章于 2025-09-06 17:22:57 发布

xiabe

最新推荐文章于 2025-09-06 17:22:57 发布

阅读量195

点赞数

CC 4.0 BY-SA版权

分类专栏： leecode 中等文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/xiabenshu/article/details/89283301

leecode 中等专栏收录该内容

19 篇文章

订阅专栏

本文介绍了一种寻找数组中第k大元素的算法，通过维护一个大小为k的列表进行排序比较，以及一种更高效的堆队列方法，后者显著提升了查找速度。

题目

在这里插入图片描述
求出数组中第k大的数字。

我的代码（维持一个k的列表）

class Solution:
    def findKthLargest(self, nums: List[int], k: int) -> int:
        d=nums[:k]
        d.sort()
        for i in range(k,len(nums)):
            if nums[i]>d[0]:
                sign=False
                for j in range(1,len(d)):
                    if nums[i]<=d[j]:
                        sign=True
                        d.insert(j,nums[i])
                        break
                if not sign:
                    d.append(nums[i])
                del d[0]
        return d[0]

高效代码

使用堆队列的方法，提高了效率。

class Solution:
    def findKthLargest(self, nums: List[int], k: int) -> int:
        heap = []
        for num in nums:
            heapq.heappush(heap, num)
            if len(heap) > k:
                heapq.heappop(heap)
        return heapq.heappop(heap)