python将一个列表调整为大根堆模板

最新推荐文章于 2024-11-29 21:53:43 发布

yuzhang_zy

最新推荐文章于 2024-11-29 21:53:43 发布

阅读量639

点赞数

CC 4.0 BY-SA版权

分类专栏：力扣 python3

本文链接：https://blog.youkuaiyun.com/qq_39445165/article/details/118570419

力扣同时被 2 个专栏收录

904 篇文章

订阅专栏

python3

76 篇文章

订阅专栏

本文介绍如何使用Python将列表转换为大根堆，通过从上至下的调整过程，确保每个父节点值大于或等于其子节点。通过实例演示了如何从中间节点开始，逐步向下进行堆化操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

有的时候需要将一个列表调整为一个大根堆或者小根堆，我们其实可以从n = len(nums) // 2的位置一直到下标为0的位置逆序开始从上到下进行调整（将数组看成是一棵完全二叉树），将列表调整为小根堆或者是大根堆其实是在循环比较当前根节点的值与左右孩子节点调整比较的符号即可，下面是将列表调整为大根堆的代码模板：

from typing import List
import random


class Solution:
    # 从上到下进行调整
    def makeToHeap(self, nums: List[int], low: int, high: int):
        i = low
        j = 2 * i + 1
        while j <= high:
            # 调整下面nums[j]与nums[j + 1]比较的符号可以调整为小根堆
            if j + 1 < len(nums) and nums[j] < nums[j + 1]: j += 1
            if nums[i] < nums[j]:
                nums[i], nums[j] = nums[j], nums[i]
                i = j
                j = 2 * i + 1
            else:
                break


if __name__ == "__main__":
    # 从控制台输入n
    n = int(input())
    nums = list()
    for i in range(n):
        # 生成0~n * 10的随机数
        nums.append(random.randint(0, n * 10))
    i = n // 2
    while i >= 0:
        # 由上到下进行调整为大根堆
        Solution().makeToHeap(nums, i, n - 1)
        i -= 1
    print(nums)