Lintcode 130 · 堆化

链接：https://www.lintcode.com/problem/130/

130 · 堆化

题目描述

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

解法

题目意思很简单，批量建堆，如果遍历数组，依次做堆的添加操作，那么时间复杂度是O(nlogn)，完全可以进行全排序了。所以在实际中，常常采用自下而上的下滤，来进行批量建堆，时间复杂度是O(n)。思路也很简单，自下而上，先做到左右子树成堆，再做到父节点也满足堆的性质，直到建堆完成。

from typing import (
    List,
)

class Solution:
    """
    @param a: Given an integer array
    @return: nothing
    """
    def heapify(self, a: List[int]):
        # write your code here
        def sift_down(index, max_len):
            non_leaf_size = max_len // 2
            while index < non_leaf_size:
                left_idx, right_idx = (index << 1) + 1, (index << 1) + 2
                if right_idx >= max_len:
                    child_idx = left_idx
                else:
                    child_idx = left_idx if a[left_idx] < a[right_idx] else right_idx
                if a[child_idx] >= a[index]:
                    break
                a[index], a[child_idx] = a[child_idx], a[index]
                index = child_idx
                
        non_leaf_size = len(a) // 2  # 非叶子节点数量
        for i in range(non_leaf_size - 1, -1, -1):  # 采用自下而上的下滤
            sift_down(i, len(a))

当然也可以进行自上而下的上滤进行建堆，但由于完全二叉树约有 n/2的叶子节点，每个叶子节点上滤又消耗log(n)的时间，为此这种方式时间复杂度为O(nlogn)，不如自下而上的下滤。