题目描述:
*给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。
请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
示例:
1)输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
2)输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
思路一:
直接对数组进行快排,返回第k大即可。
之前在牛客上做过,对应的[链接]为:(https://blog.youkuaiyun.com/coding_diamond/article/details/121349922)
或者:
class Solution(object):
def findKthLargest(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: int
"""
a = sorted(nums)
return a[len(nums)-k]
思路二:
直接利用堆返回前K大,取前K大的数组的最后一个。
class Solution(object):
def findKthLargest(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: int
"""
a = heapq.nlargest(k, nums)
return a[-1]
思路三:
与流的问题一样,维持一个尺寸为K的小顶堆,然后不断加入元素,维护这个小顶堆,这样几种方法的时间复杂相同的。
代码可以参考:这个链接。