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、数据流的中位数