[leetcode/lintcode 题解]算法面试真题详解:堆化

描述
给出一个整数数组,堆化操作就是把它变成一个最小堆数组。
对于堆数组A,A[0]是堆的根,并对于每个A[i],A [i 2 + 1]是A[i]的左儿子并且A[i 2 + 2]是A[i]的右儿子。

在线评测地址:领扣题库官网

说明
什么是堆?什么是堆化?如果有很多种堆化的结果?

  • 堆是一种数据结构,它通常有三种方法:push, pop 和 top。其中,“push”添加新的元素进入堆,“pop”删除堆中最小/最大元素,“top”返回堆中最小/最大元素。
  • 把一个无序整数数组变成一个堆数组。如果是最小堆,每个元素A[i],我们将得到A[i 2 + 1] >= A[i]和A[i 2 + 2] >= A[i]
  • 返回其中任何一个。
样例
输入 : [3,2,1,4,5]
输出 : [1,2,3,4,5]
解释 : 返回任何一个合法的堆数组

Heapify 的具体实现方法。时间复杂度为O(n)O(n),使用的是 siftdown 之所以是 O(n) 是因为从第 N/2 个位置开始往下 siftdown,那么就有大约 N/4 个数在 siftdown 中最多交换 1 次,N/8 个数最多交换 2 次,N/16 个数最多交换 3 次。 所以O(N/4∗1+N/8∗2+N/16∗3+...+1∗LogN)=O(N)

class Solution:
    """
    @param: A: Given an integer array
    @return: nothing
    """
    def heapify(self, A):
        for i in range(len(A) // 2, -1, -1):
            self.siftdown(A, i)
            
    def siftdown(self, A, index):
        n = len(A)
        while index < n:
            left = index * 2 + 1
            right = index * 2 + 2
            minIndex = index
            if left < n and A[left] < A[minIndex]:
                minIndex = left
            if right < n and A[right] < A[minIndex]:
                minIndex = right

            if minIndex == index:
                break
            
            A[minIndex], A[index] = A[index], A[minIndex]
            index = minIndex

更多题解参考:九章官网solution

上一篇:[leetcode/lintcode 题解] 阿里算法面试真题:最长有效括号


下一篇:[leetcode/lintcode 题解] 阿里面试题:字模式II