|
| 1 | +#================================================== |
| 2 | +#==> Title: 215. 数组中的第K个最大元素 |
| 3 | +#==> Author: Zhang zhen |
| 4 | +#==> Email: hustmatnoble.gmail.com |
| 5 | +#==> GitHub: https://github.com/MatNoble |
| 6 | +#==> Date: 2/13/2021 |
| 7 | +#================================================== |
| 8 | + |
| 9 | +""" |
| 10 | +https://leetcode-cn.com/problems/kth-largest-element-in-an-array/ |
| 11 | +""" |
| 12 | + |
| 13 | +from typing import List |
| 14 | +class Solution: |
| 15 | + def findKthLargest(self, nums: List[int], k: int) -> int: |
| 16 | + self.k = len(nums) - k |
| 17 | + def partition(array, left, right): |
| 18 | + # 随机选取 pivot, 并交换至 right |
| 19 | + idx = random.randint(left, right) |
| 20 | + array[idx], array[right] = array[right], array[idx] |
| 21 | + pivot = array[right] |
| 22 | + i, j= left, right - 1 |
| 23 | + while True: |
| 24 | + while i < right and array[i] <= pivot: i += 1 |
| 25 | + while j >= left and array[j] > pivot: j -= 1 |
| 26 | + if i > j: break |
| 27 | + array[i], array[j] = array[j], array[i] |
| 28 | + array[right], array[i] = array[i], array[right] |
| 29 | + return i |
| 30 | + def quickQort(array, left, right): |
| 31 | + if left >= right: return |
| 32 | + ## 前序遍历 |
| 33 | + index = partition(array, left, right) |
| 34 | + if self.k == index: |
| 35 | + return |
| 36 | + elif self.k < index: |
| 37 | + quickQort(array, left, index-1) |
| 38 | + else: |
| 39 | + quickQort(array, index+1, right) |
| 40 | + quickQort(nums, 0, len(nums)-1) |
| 41 | + return nums[self.k] |
| 42 | + |
| 43 | +mat = Solution() |
| 44 | +nums = [3,2,1,5,6,4] |
| 45 | +k = 3 |
| 46 | +mat.findKthLargest(nums, k) |
0 commit comments