在数组中找到第k大的元素 样例 给出数组 [9,3,2,4,8],第三大的元素是 4 给出数组 [1,2,3,4,5],第一大的元素是 5,第二大的元素是 4,第三大的元素是 3,以此类推 挑战 要求时间复杂度为O(n),空间复杂度为O(1),设函数为 find(n,nums)nums为给定数组,n是指找到第n大的元素
此题可以采用快速选择来解。
找到数组nums[]中间元素nums[mid],然后创建两个数组big[] 和small[]分别存放大于nums[mid]和小于nums[mid]的元素然后重复以下过程:
1.如果big数组的长度等于n-1,那么说明nums[mid]为第n大的元素,return nums[mid]
2如果big数组的长度大于n-1,那么说明第n大的元素在数组big中,执行 return find(n,big)
3.如果big数组长度小于n-1,那么说明第n大的元素在数组small中,执行return find(n-len(big)-1-num,small)
这里的num是用来处理重复的nums[mid]的
def find(n,nums):
if n > len(nums) or n < 1 or nums == None:
return None
mid = len(nums)/2
big = []
small = []
num = 0
for i in range(len(nums)):
if nums[i] > nums[mid]:
big.append(nums[i])
if nums[i] < nums[mid]:
small.append(nums[i])
if nums[i] == nums[mid]:
num += 1
num -= 1
if len(big) == n - 1:
return nums[mid]
if len(big) > n - 1:
return find(n,big)
if len(big) < n - 1:
if len(big) + num >= n-1:
return nums[mid]
else:
return find(n-len(big)-1-num,small)