[LeetCode]215. Kth Largest Element in an Array

最新推荐文章于 2025-09-05 14:01:23 发布

原创最新推荐文章于 2025-09-05 14:01:23 发布 · 122 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #排序 #python #python3

leetcode 专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种有效的方法来找到未排序数组中的第k个最大元素。该方法使用了快速选择算法的思想，通过部分排序的方式避免了完全排序整个数组，从而提高了效率。

找到数组中第k大的数
215. Kth Largest Element in an Array

class Solution:
    def findKthLargest(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        target = len(nums) - k
        low, high = 0 , len(nums)-1
        while low < high:
            l,r = low, high
            p = nums[l]
            while l < r:
                while l < r and nums[r] >= p:
                    r -= 1
                nums[l] = nums[r]
                while l < r and nums[l] <= p:
                    l += 1
                nums[r] = nums[l]
            nums[l] = p
            if l == target:
                return p
            elif l < target:
                low, high = l+1, high
            else:
                low, high = low, l-1
        return nums[low]