题目来源:Leetcode 215 数组中第K个最大的元素
题目描述:
在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
示例 1:
输入: [3,2,1,5,6,4] 和
k = 2
输出: 5
1.排序方法
复杂度为O(nlogn),可以直接调用python的排序函数做(用时52ms):
class Solution:
def findKthLargest(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: int
"""
nums.sort()
return nums[-k]
自己手写快排(用时284ms,咋这么慢。。。):
class Solution:
def findKthLargest(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: int
"""
import random
self.quickSort(nums, 0, len(nums) - 1)
return nums[-k]
def partation(self, nums, l, r):
"""随机partation"""
change_index = random.randint(l, r)
nums[change_index], nums[r] = nums[r], nums[change_index]
num = nums[r]
less_index = l - 1
more_index = r + 1
p = l
while p < more_index:
if nums[p] < num:
less_index += 1
nums[p], nums[less_index] = nums[less_index], nums[p]
p +&#