数组中的第K个最大元素

https://leetcode-cn.com/explore/interview/card/bytedance/243/array-and-sorting/1018/

在未排序的数组中找到第 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 总是有效的,且 1 ≤ k ≤ 数组的长度。

数组中的第K个最大元素,这个题难度比较适中,可以检验下自己排序的基础知识是否扎实。

比较好的思路:借鉴快速排序,快速排序是选一个pos,然后比pos小的交换到左边,pos大的交换到右边
交换一轮后就可以找到pos是k_pos几大的元素(注意第几大是指倒序排序!)。
每次快排完成后,有3种情况:

  1. k_pos == target:恭喜直接返回pos
  2. k_pos>target: k_pos是指数组中第k_pos的数,比如pos是第10大的数,但是我要找target是第5大的数,说明要找的数还在我的右边,同时target不用变。
  3. k_pos<target: 例如pos是第10大的数,我要找第15大的数,说明得在pos的左边继续找,而且只需要找第5大的数就好啦
    代码如下:
class Solution(object):
    def findKthLargest(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        if not nums:
            return None
        left = 0
        right = len(nums) - 1
        while left <= right:
            pos = self.quicksort(nums, left, right)
            k_sort = right-pos+1 # 比pos大的数字个数,比如pos是第k_sort大的数
            if k_sort == k:
                return nums[pos]
            elif k_sort > k:
                # 比pos大的个数k,说明还需要右边继续找
                left = pos + 1
            else:
                k = k - k_sort
                right = pos - 1
        return None

    def quicksort(self, nums, left, right):
        pos = nums[left]
        start = left
        while left != right:
            while (nums[right] >= pos and left < right):
                right -= 1
            while (nums[left] <= pos and left < right):
                left += 1
            if left != right:
                nums[left], nums[right] = nums[right], nums[left]
            else:
                nums[left], nums[start] = nums[start], nums[left]
        return left

第二个思路也很直接,构建一个长度为K的小跟堆(注意是小跟堆!)
然后堆顶其实就是第K大的数,循环这个数组,迭代这个小跟堆,最后返回堆顶元素就好了。这个思路比较考验手写堆排序,堆排序值得注意的一点是,根节点为i,左节点为i2+1,右节点坐标为i2+2。在调整堆的遍历中,是要满足左节点在范围内既要遍历,所以是i*2+1<=end。具体代码如下:

class Solution(object):
    def findKthLargest(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        if not nums:
            return None
        if len(nums) ==1 and k==1:
            return nums[0]
        heap_nums = []
        for i in range(0,k):
            heap_nums.append(nums[i])

       # 建立K的小跟堆
        for i in range(k/2-1,-1,-1):
            self.adjustHeap(heap_nums,i,k-1)
      # 遍历数组,找到第K大的数字,如果发现元素比堆顶大,则要更新堆了
        for i in range(k,len(nums)):
            if nums[i] > heap_nums[0]:
                heap_nums[0] = nums[i]
                self.adjustHeap(heap_nums,0,k-1)
        return heap_nums[0]
            
        
    def adjustHeap(self,heap_nums,start,end):
        pos = start
        while pos <= (end-1)/2:
            min_child_pos = pos*2+1 # 默认设置为left值
            # 找到左右节点中最小的节点
            right_child_pos = pos*2 +2
            if right_child_pos <= end:
                if heap_nums[min_child_pos]>heap_nums[right_child_pos]:
                    min_child_pos = right_child_pos
            
            if heap_nums[pos]>heap_nums[min_child_pos]:
                heap_nums[pos],heap_nums[min_child_pos] = heap_nums[min_child_pos],heap_nums[pos]
                pos = min_child_pos
            else:
                break
        return 
                    
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,591评论 6 501
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,448评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,823评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,204评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,228评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,190评论 1 299
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,078评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,923评论 0 274
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,334评论 1 310
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,550评论 2 333
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,727评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,428评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,022评论 3 326
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,672评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,826评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,734评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,619评论 2 354