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

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

• 题目：给定整数数组 nums 和整数 k，请返回数组中第 k 个最大的元素。
• 示例：输入: [3,2,1,5,6,4] 和 k = 2 输出: 5

使用快速排序实现时间复杂度为O(n)，空间复杂度为O(logn)的算法：

class Solution:
    def findKthLargest(self, nums: List[int], k: int) -> int:

        def partPos( arr , left , right ):
            #初始化最右端为比较数
            pivot = arr[right]
            i = left
            for j in range(left,right+1):
                #将所有arr[j]<=pivot的数字与arr[i]交换，i+1
                if arr[j]<=pivot:
                    arr[i],arr[j] = arr[j],arr[i]
                    i += 1
            #返回当前pivot位置i-1，此时i-1左右被pivot分开
            return i-1

        def quickSort( arr , left , right , distance ):
            #一次快排，返回中间位置
            pos = partPos( arr , left , right )
            #当前位置距离right==distance立即返回，否则递归查找
            if right-pos+1 == distance:
                return arr[pos]
            elif right-pos+1 < distance:
                return quickSort( arr , left , pos-1 , distance-(right-pos+1) )
            elif right-pos+1 > distance:
                return quickSort( arr , pos+1 , right , distance )
        
        return quickSort( nums , 0 , len(nums)-1 , k )