lintcode: 堆化

Posted 2020-06-25

 堆化

给出一个整数数组，堆化操作就是把它变成一个最小堆数组。

对于堆数组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] 或者任何一个合法的堆数组

解题

根据给的样例，直接排序后就符合答案。

public class Solution {
    /**
     * @param A: Given an integer array
     * @return: void
     */
    public void heapify(int[] A) {
        // write your code here
        Arrays.sort(A);
    }
}

参考链接 

递归进行堆排序

public class Solution {
    /**
     * @param A: Given an integer array
     * @return: void
     */
    public void heapify(int[] A) {
        // write your code here
        for(int i = (A.length - 1)/2;i>=0;i--)
            heapify(A,i);
    }
    public void heapify(int[] A,int i){
        int l = 2*i + 1;
        int r = 2*i + 2;
        int smallest = i;
        if(l < A.length && A[l] < A[smallest])
            smallest = l;
        if( r < A.length && A[r] < A[smallest])
            smallest = r;
        if(smallest!=i){
            int tmp = A[i];
            A[i] = A[smallest];
            A[smallest] = tmp;
            heapify(A,smallest);
        }
    }
}