堆排序

4 篇文章 0 订阅
3 篇文章 0 订阅

堆排序的过程:

1.创建初始堆。

2.将堆的最后一个元素和堆顶元素交换,减少堆的范围, 调整堆。

class Solution:
    def findKthLargest(self, nums: List[int], k: int) -> int:
        len_nums = len(nums)
        for i in range(len_nums//2-1, -1, -1):
            self.adjust_heap(nums, i, len_nums) # 从最后一个非叶子节点向上调整堆
        for j in range(len_nums-1, 0, -1):
            nums[0], nums[j] = nums[j], nums[0] # 交换堆顶元素和堆最后一个元素
            self.adjust_heap(nums, 0, j)        # 缩小堆的范围,调整堆顶
        return nums
    def adjust_heap(self, target_list, i, target_len):
        k = 2 * i + 1 # 左孩子坐标
        tmp = target_list[i]
        while k < target_len:
            if k+1 < target_len and target_list[k+1] > target_list[k]:
                k += 1
            if tmp < target_list[k]: # 需要和tmp比,因为i在变化
                target_list[i] = target_list[k]
                i = k
            else: # 根据堆的性质,后面的不需要继续比了
                break
            k = 2 * k + 1
        target_list[i] = tmp

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值