python leetcode 215. Kth Largest Element in an Array

最新推荐文章于 2021-04-16 21:50:05 发布

Neekity

最新推荐文章于 2021-04-16 21:50:05 发布

阅读量643

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode python 文章标签： python leetcode

本文链接：https://blog.youkuaiyun.com/Neekity/article/details/85048034

python 同时被 2 个专栏收录

263 篇文章

订阅专栏

leetcode

221 篇文章

订阅专栏

本文介绍了一种使用堆排序算法寻找数组中第K大的元素的方法。通过将数组转换为一个最大堆，然后进行K次弹出操作，可以高效地找到所需的元素。这种方法在大数据集上比传统排序更有效。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

堆排序

class Solution:
    def findKthLargest(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        import heapq
        nums = [-num for num in nums]
        heapq.heapify(nums)
        res = float('inf')
        for _ in range(k):
            res = heapq.heappop(nums)
        return -res