215. Kth Largest Element in an Array

ujn20161222

已于 2022-05-27 04:23:57 修改

阅读量76

点赞数

文章标签： python

于 2022-03-22 06:59:45 首次发布

本文链接：https://blog.csdn.net/ujn20161222/article/details/123650889

版权

这篇博客介绍了一种解决编程问题的方法，即如何在给定的整数数组中找到第K大的元素。算法采用了随机化选择枢轴的快速选择策略，通过递归地将数组分为小于、等于和大于枢轴的三部分，逐步缩小搜索范围。当K位于左右子数组的大小范围内时，返回相应部分的元素。这种方法避免了完全排序，降低了时间复杂度。

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

215. Kth Largest Element in an Array

Medium

8621481Add to ListShare

Given an integer array nums and an integer k, return the kth largest element in the array.

Note that it is the kth largest element in the sorted order, not the kth distinct element.

Example 1:

Input: nums = [3,2,1,5,6,4], k = 2
Output: 5

Example 2:

Input: nums = [3,2,3,1,2,4,5,5,6], k = 4
Output: 4

Constraints:

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

Accepted

1,245,463

Submissions

1,976,476

import random
class Solution:
    def findKthLargest(self, nums: List[int], k: int) -> int:
        if not nums: return
        pivot=random.choice(nums)
        left=[x for x in nums if x>pivot]
        mid=[x for x in nums if x==pivot]
        right=[x for x in nums if x<pivot]
        
        L,M=len(left),len(mid)
        
        if k<=L:
            return self.findKthLargest(left,k)
        elif k>L+M:
            return self.findKthLargest(right,k-L-M)
        else:
            return mid[0]

每次会drop掉非常大的一部分

random pivot

注意左边存的大数