LeetCode:215. Kth Largest Element in an Array

XP-Code

于 2019-07-23 15:07:36 发布

阅读量160

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/Wisimer/article/details/96996594

LeetCode 专栏收录该内容

58 篇文章

订阅专栏

本文介绍了LeetCode上第215题“数组中的第K个最大元素”的两种解决方案：一种是通过排序来找到第K大的元素，另一种则是利用大顶堆实现。这两种方法各有优劣，适用于不同场景。

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

LeetCode:215. Kth Largest Element in an Array

Find the kth largest element in an unsorted array. 
Note that it is the kth largest element in the sorted order, not the kth distinct element.

Example 1:

Input: [3,2,1,5,6,4] and k = 2
Output: 5
Example 2:

Input: [3,2,3,1,2,4,5,5,6] and k = 4
Output: 4

思路一：排序

先排序，再找出第k大的数。

Python 实现代码

class Solution:
    def findKthLargest(self, nums: List[int], k: int) -> int:
        res = sorted(nums)[::-1]
        return res[k-1]

思路二：大顶堆

直接使用大顶堆找到topK的值，取最后一个就是第k大的值。

Python 实现代码

class Solution:
    def findKthLargest(self, nums: List[int], k: int) -> int:
        topK = heapq.nlargest(k, nums)
        return topK[-1]

THE END.