1、从大到小第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个数。首先想到的是对数组排序,直接sort()取k-1个数。复杂度为
O(NlogN).
方法二:根据快排中partition的思想,进行分治,,存在情况1:[left,k,p,right] 情况2:[left,p,k,right]
对于情况1继续遍历[left,p],情况2继续遍历[p,right].
class Solution(object):
def findKthLargest(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: int
"""
#partition
def core(left,right,nums,k):
r = nums[right]
i = left
little = right-1
while i<=little:
if nums[i]>=r:
i+=1
else:
#交换i,little
self.swap(i,little,nums)
little-=1
self.swap(i,right,nums)
if i==k-1:
return nums[i]
elif i<k-1:
return core(i+1,right,nums,k)
else:
return core(left,i-1,nums,k)
return core(0,len(nums)-1,nums,k)
def swap(self,i,j,nums):
tmp = nums[i]
nums[i] = nums[j]
nums[j] = tmp
2、从大到小第K大的数
3、数据流的中位数