LintCode-堆化

标签：lintcode   面试   

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

对于堆数组A，A[0]是堆的根，并对于每个A[i]，A [i * 2 + 1]是A[i]的左儿子并且A[i * 2 + 2]是A[i]的右儿子。

分析：一开始想到堆化么肯定就是堆排序吧，粗粗一想貌似复杂度是O(nlgn)，后来参考该文章才知道O(nlgn)是复杂度上限，平均是O(n)

代码：

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

LintCode-堆化

标签：lintcode   面试   

原文地址：http://blog.csdn.net/wangyuquanliuli/article/details/45826193