品易云推流 关闭
文章详情页
文章 > Python基础教程 > python数据结构堆的介绍

python数据结构堆的介绍

python数据结构

头像

小妮浅浅

2021-05-27 17:05:531056浏览 · 0收藏 · 0评论

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数据结构堆的介绍,希望对大家有所帮助。更多Python学习指路:python基础教程

关注

关注公众号,随时随地在线学习

本教程部分素材来源于网络,版权问题联系站长!

底部广告图