nowcoder 堆排序

最新推荐文章于 2024-07-31 13:26:59 发布

原创最新推荐文章于 2024-07-31 13:26:59 发布 · 205 阅读

0 ·

CC 4.0 BY-SA版权

OJ-nowcoder 同时被 2 个专栏收录

70 篇文章

订阅专栏

算法-排序

22 篇文章

订阅专栏

本文介绍如何实现堆排序算法，通过建立大根堆，将第一个元素与末尾元素交换，并不断调整堆，以此对给定的整数数组进行排序。提供了具体的测试样例和代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

对于一个int数组，请编写一个堆排序算法，对数组元素排序。
给定一个int数组A及数组的大小n，请返回排序后的数组。
测试样例：
[1,2,3,5,2,3],6
[1,2,2,3,3,5]

思路

堆排序，建立一个大根堆，然后交换第一个和最后一个元素，不断调整堆。

代码

class HeapSort:
    def adjustHeap(self, A, parent, n):
        while parent < n:
            child1 = 2 * parent
            child2 = 2 * parent + 1
            if child1 < n:
                maxchild = child1
                if child2 < n:
                    maxchild = child1 if A[child1] > A[child2] else child2
                if A[parent] < A[maxchild]:
                    A[parent], A[maxchild] = A[maxchild], A[parent]
                    parent = maxchild
                else:
                    break
            else:
                break
    def heapSort(self, A, n):
        # write code here
        for i in range(n // 2 + 1)[::-1]:
            self.adjustHeap(A, i, n)
        for i in range(n)[::-1]:
            A[0], A[i] = A[i], A[0]
            self.adjustHeap(A, 0, i)
        return A