在未排序的數(shù)組中找到第 k 個最大的元素土至。請注意购对,你需要找的是數(shù)組排序后的第 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
說明:
你可以假設(shè) k 總是有效的骡苞,且 1 ≤ k ≤ 數(shù)組的長度。
思路--深度優(yōu)先遍歷
python3解法
class Solution:
def findKthLargest(self, nums: List[int], k: int) -> int:
# 使用小頂堆
q = []
for c in nums:
# heapq模塊提供了堆排序算法的實現(xiàn)
# 創(chuàng)建heapq:使用一個空列表楷扬,然后使用heapq.heappush()函數(shù)把值加入堆中
heapq.heappush(q, c)
# 當(dāng)q的長度大于k時解幽,就講最小值彈出
# 遍歷完nums之后,q中剩余k個數(shù)烘苹,最小值就是第k個最大的元素
while len(q) > k:
heapq.heappop(q)
return heapq.heappop(q)
來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/kth-largest-element-in-an-array
著作權(quán)歸領(lǐng)扣網(wǎng)絡(luò)所有躲株。商業(yè)轉(zhuǎn)載請聯(lián)系官方授權(quán),非商業(yè)轉(zhuǎn)載請注明出處镣衡。