编程语言
首页 > 编程语言> > python将一个列表调整为大根堆模板

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

作者:互联网

有的时候需要将一个列表调整为一个大根堆或者小根堆,我们其实可以从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)

标签:nums,python,根堆,int,low,列表,模板,调整
来源: https://blog.csdn.net/qq_39445165/article/details/118570419