python數據結構堆的示例分析

小編給大家分享一下python數據結構堆的示例分析，希望大家閱讀完這篇文章之后都有所收獲，下面讓我們一起去探討吧！

1、說明

堆是用數據結構來實現的一種算法：樹，數組均可。堆本身是一棵完全二叉樹。

2、特點

最大堆：所有父節點的值大于子節點的值

最小堆：所有父節點的值小于子節點的值

3、實例

class Heap(object):
    def __init__(self, list=[]):
        self.root = None
        self.list = list
        self.tree = None
        self.len = len(list)
 
    # 建堆
    def bulid_heap(self):
        if self.list != []:
            final_parent_node = int((self.len - 1) / 2)
            while final_parent_node >= 0:
                self.heapfy(final_parent_node, self.len)
                final_parent_node -= 1
 
    # 對當前節點以及向下所有子節點的一次最大節點交換
    def heapfy(self, node, len):
        node_left = 2 * node + 1
        node_right = 2 * node + 2
        max = node
        if node_left < len and self.list[node_left] > self.list[max]:
            max = node_left
        if node_right < len and self.list[node_right] > self.list[max]:
            max = node_right
        if max != node:
            self.swap(max, node)
            self.heapfy(max, len)
 
    # 交換元素方法
    def swap(self, i, j):
        self.list[j], self.list[i] = self.list[i], self.list[j]
 
    # 堆排序
    def heap_sort(self):
        len = self.len - 1
        while len >= 0:
            self.swap(0, len)
            self.heapfy(0, len)
            len -= 1
 
 
if __name__ == "__main__":
    list = [5, 7, 3, 1, 10, 0]
    heap = Heap(list)
    print("初始列表：{}".format(heap.list))
    heap.bulid_heap()
    print("堆化：{}".format(heap.list))
    heap.heap_sort()
    print("排序：{}".format(heap.list))

看完了這篇文章，相信你對“python數據結構堆的示例分析”有了一定的了解，如果想了解更多相關知識，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！