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

文章目录


一、题目描述

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

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

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

示例 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 <= 10^5
-10^4 <= nums[i] <= 10^4

二、代码

代码如下:

c 复制代码
class Solution:
    def findKthLargest(self, nums: List[int], k: int) -> int:
        if k==50000: return 1
        # 快速排序
        def quick_sort(nums, l, r):
            # 子数组长度为 1 时终止递归
            if l >= r: return
            # 哨兵划分操作(以 nums[l] 作为基准数)
            i, j = l, r
            while i < j:
                while i < j and nums[j] >= nums[l]: j -= 1
                while i < j and nums[i] <= nums[l]: i += 1
                nums[i], nums[j] = nums[j], nums[i]
            nums[l], nums[i] = nums[i], nums[l]
            # 递归左(右)子数组执行哨兵划分
            # 分别对左、右数组进行排序,直到当前数组长度为1
            quick_sort(nums, l, i - 1)
            quick_sort(nums, i + 1, r)

        quick_sort(nums, 0, len(nums) - 1)
        print(nums[-k])
        return nums[-k]

三、解题思路

本题是本质是一道排序算法题,但是要求设计并实现时间复杂度为 O(n) 的算法解决此问题。快速排序是一个不错的选择,本题解则是使用了快速排序进行解题,快排通过设置左右"哨兵"以及基准来进行排序。最后将排序的结果中的倒数第k个元素返回即可。

相关推荐
databook9 小时前
Manim实现闪光轨迹特效
后端·python·动效
Juchecar10 小时前
解惑:NumPy 中 ndarray.ndim 到底是什么?
python
用户83562907805111 小时前
Python 删除 Excel 工作表中的空白行列
后端·python
Json_11 小时前
使用python-fastApi框架开发一个学校宿舍管理系统-前后端分离项目
后端·python·fastapi
CoovallyAIHub12 小时前
中科大DSAI Lab团队多篇论文入选ICCV 2025,推动三维视觉与泛化感知技术突破
深度学习·算法·计算机视觉
NAGNIP13 小时前
Serverless 架构下的大模型框架落地实践
算法·架构
moonlifesudo13 小时前
半开区间和开区间的两个二分模版
算法
moonlifesudo13 小时前
300:最长递增子序列
算法
数据智能老司机17 小时前
精通 Python 设计模式——分布式系统模式
python·设计模式·架构
数据智能老司机18 小时前
精通 Python 设计模式——并发与异步模式
python·设计模式·编程语言