Python实现堆排序算法,轻松掌握堆排序的原理和实现

更新时间:2024-06-18 分类:网络技术 浏览量:2

什么是堆排序

在计算机科学中,堆排序是一种比较排序算法,它使用堆这种特殊的数据结构来进行排序。堆是一个完全二叉树,分为最大堆和最小堆两种形式,它具有“堆积”的特性,父节点的键值总是大于或等于子节点(最大堆)或者小于或等于子节点(最小堆)。

堆排序的原理

堆排序的基本思想是将待排序序列构造成一个堆,然后将堆顶元素和堆底元素交换,然后调整堆结构使其满足堆的定义,再重复进行堆顶元素和堆底元素交换及调整堆的操作直到整个序列有序。

Python实现堆排序的代码

下面是Python实现堆排序的代码:

    
    def heapify(arr, n, i):
        largest = i  
        l = 2 * i + 1     
        r = 2 * i + 2     
     
        if l < n and arr[i] < arr[l]:
            largest = l
     
        if r < n and arr[largest] < arr[r]:
            largest = r
     
        if largest != i:
            arr[i], arr[largest] = arr[largest], arr[i]
            heapify(arr, n, largest)
     
    def heapSort(arr):
        n = len(arr)
     
        for i in range(n // 2 - 1, -1, -1):
            heapify(arr, n, i)
     
        for i in range(n-1, 0, -1):
            arr[i], arr[0] = arr[0], arr[i]
            heapify(arr, i, 0)
     
    arr = [12, 11, 13, 5, 6, 7]
    heapSort(arr)
    n = len(arr)
    print("排序后的数组为")
    for i in range(n):
        print("%d" % arr[i])
    
    

堆排序的应用

堆排序由于其平均时间复杂度为O(nlogn)且不占用额外空间而被广泛应用于实际工程中。

希望通过本文你可以轻松掌握堆排序的原理和Python实现代码,为你的排序算法知识储备添砖加瓦。感谢阅读!