LeetCode 215:数组中的第k大元素

这篇博客介绍了三种方法来找到整数数组中第k个最大的元素。方法一使用快速排序,方法二通过堆排序,方法三利用小顶堆。每种方法都有其适用场景,详细代码和解释帮助理解算法的运作过程。

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

题目描述:
*给定整数数组 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的小顶堆,然后不断加入元素,维护这个小顶堆,这样几种方法的时间复杂相同的。
代码可以参考:这个链接

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值