老卫带你学---leetcode刷题(215. 数组中的第K个最大元素)

215. 数组中的第K个最大元素

问题:

给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。

请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。

你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。

bash 复制代码
示例 1:

输入: [3,2,1,5,6,4], k = 2
输出: 5
bash 复制代码
示例 2:

输入: [3,2,3,1,2,4,5,5,6], k = 4
输出: 4
bash 复制代码
提示:

1 <= k <= nums.length <= 105
-104 <= nums[i] <= 104

解决:

堆排序

对每个元素入堆,然后pop出来k-1个

这里需要注意,默认堆为最小堆,所以在入堆的时候入num的负数,就会trick为最大堆

python 复制代码
import heapq
class Solution:
    def findKthLargest(self, nums: List[int], k: int) -> int:
        l = []
        for num in nums:
            heapq.heappush(l,-num)
            print(l)
        for _ in range(k-1):
            heapq.heappop(l)
        return -l[0]
相关推荐
科大饭桶4 小时前
数据结构自学Day5--链表知识总结
数据结构·算法·leetcode·链表·c
YuTaoShao7 小时前
【LeetCode 热题 100】24. 两两交换链表中的节点——(解法一)迭代+哨兵
java·算法·leetcode·链表
前端拿破轮8 小时前
翻转字符串里的单词,难点不是翻转,而是正则表达式?💩💩💩
算法·leetcode·面试
凤年徐8 小时前
【数据结构与算法】203.移除链表元素(LeetCode)图文详解
c语言·开发语言·数据结构·算法·leetcode·链表·刷题
李昊_8 小时前
【LeetCode 3440. 重新安排会议得到最多空余时间 II】解析
算法·leetcode
呆呆的小鳄鱼9 小时前
leetcode:322. 零钱兑换[完全背包]
算法·leetcode·职场和发展
এ᭄画画的北北9 小时前
力扣-240.搜索二维矩阵 II
算法·leetcode·矩阵
JJ1M810 小时前
前缀和+贪心总结,基于每日一题力扣3439、3440
python·算法·leetcode
呆呆的小鳄鱼10 小时前
leetcode:518. 零钱兑换 II[完全背包]
算法·leetcode·职场和发展
沧澜sincerely11 小时前
二分查找【各种题型+对应LeetCode习题练习】
算法·leetcode·二分查找