剑指 Offer II 076. 数组中的第 k 大的数字

最新推荐文章于 2024-10-11 12:01:25 发布

qbnmm

最新推荐文章于 2024-10-11 12:01:25 发布

阅读量201

点赞数

分类专栏：刷题笔记文章标签：算法 leetcode

本文链接：https://blog.csdn.net/qq_33523932/article/details/125704519

版权

刷题笔记专栏收录该内容

118 篇文章 1 订阅

订阅专栏

该博客介绍了如何利用Python的heapq库解决LeetCode中的找到数组中第K大的元素问题。通过堆排序的方法，首先将数组元素插入到最小堆中，然后依次弹出堆顶元素，最终得到第K大的元素。代码简洁高效，适用于处理大规模数据。示例展示了在不同输入情况下的正确输出。

摘要由CSDN通过智能技术生成

题目链接

力扣

题目描述

给定整数数组 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

提示：

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

解题思路

堆排序

代码

Python

class Solution:
    def findKthLargest(self, nums: list[int], k: int) -> int:
        res = []
        for num in nums:
            heapq.heappush(res, -num)
        for i in range(k-1):
            heapq.heappop(res)
        return -heapq.heappop(res)