【215】第K个大的数，K相关题目

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

标签：little,215,题目,nums,self,right,从大到,相关,left
来源： https://blog.csdn.net/weixin_39530880/article/details/118097021