leetcode 215. 数组中的第K个最大元素(python)

215. 数组中的第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 总是有效的，且 1 ≤ k ≤ 数组的长度。

思路：快排 分区的思想 然后再判断

class Solution:
    def findKthLargest(self, nums, k: int) -> int:
        if not nums:
            return None
        low , high = 0 , len(nums)-1
        while True:
            pos = self.partition(nums, low, high)
            if pos == len(nums)-k:
                return nums[pos]
            elif pos > len(nums)-k:
                high = pos -1
            else:
                low = pos +1
    def partition(self,nums, low, high):
        pivot = nums[low]
        while low < high:
            while low < high and pivot <= nums[high]:
                high -= 1
            nums[low] = nums[high]
            while low < high and pivot > nums[low]:
                low += 1
            nums[high] = nums[low]
        nums[low] = pivot
        return low

len(nums) - k :表示第K大的数。
k-1 :表示第K小的数。

进阶题 最小的K个数 / 最大的K个数