中文字幕av专区_日韩电影在线播放_精品国产精品久久一区免费式_av在线免费观看网站

溫馨提示×

Python TreeNode如何實現樹的堆排序

小樊
81
2024-08-17 16:25:40
欄目: 編程語言

實現樹的堆排序可以使用Python中的TreeNode類來表示樹節點,同時使用堆排序算法來對樹進行排序。以下是一個示例代碼:

class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right

def heapify(root, size, i):
    largest = i
    l = 2 * i + 1
    r = 2 * i + 2
    
    if l < size and root[l].val > root[largest].val:
        largest = l
    
    if r < size and root[r].val > root[largest].val:
        largest = r
    
    if largest != i:
        root[i], root[largest] = root[largest], root[i]
        heapify(root, size, largest)

def heap_sort(root):
    size = len(root)
    
    for i in range(size//2 - 1, -1, -1):
        heapify(root, size, i)
    
    for i in range(size-1, 0, -1):
        root[i], root[0] = root[0], root[i]
        heapify(root, i, 0)
    
    return root

# 示例
root = [TreeNode(4), TreeNode(10), TreeNode(7), TreeNode(5), TreeNode(1)]

sorted_root = heap_sort(root)

for node in sorted_root:
    print(node.val)

在上面的代碼中,我們首先定義了TreeNode類來表示樹節點,然后實現了heapify函數來維護堆的性質,以及heap_sort函數來對樹進行堆排序。最后我們通過示例代碼對一個樹節點列表進行堆排序,并輸出排序后的結果。

0
观塘区| 阳江市| 托克托县| 海口市| 葵青区| 遵义县| 云南省| 登封市| 长海县| 武山县| 曲靖市| 卫辉市| 吉木萨尔县| 九龙县| 武义县| 射洪县| 彰武县| 休宁县| 仁怀市| 塔城市| 鹤岗市| 保靖县| 大荔县| 天水市| 苏尼特左旗| 新野县| 微山县| 永福县| 贵州省| 文昌市| 米易县| 韶山市| 淮安市| 黑龙江省| 历史| 石嘴山市| 博乐市| 永善县| 大冶市| 肥东县| 措美县|