130-堆化
给出一个整数数组,堆化操作就是把它变成一个最小堆数组。
对于堆数组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] 或者任何一个合法的堆数组
挑战
O(n)的时间复杂度完成堆化
标签
堆 LintCode 版权所有
思路
采用堆排序
code
class Solution {
public:
/**
* @param A: Given an integer array
* @return: void
*/
void heapify(vector<int> &A) {
// write your code here
int size = A.size();
if(size <= 0) {
return;
}
for(int i=(size-1)/2; i>=0; i--) {
heapify(A, i);
}
}
void heapify(vector<int> &A, int i) {
int leftChild = 2 * i + 1;
int rightChild = 2 * i + 2;
int minNode = i;
if(leftChild < A.size() && A[leftChild] < A[minNode]) {
minNode = leftChild;
}
if(rightChild < A.size() && A[rightChild] < A[minNode]) {
minNode = rightChild;
}
if(minNode != i) {
swap(A[i], A[minNode]);
heapify(A, minNode);
}
}
};