基本思路
- 1.若array[0,…,n-1]表示一颗完全二叉树的顺序存储模式,则双亲节点指针和孩子结点指针之间的内在关系如下:
任意一节点指针 i:
父节点:i==0 ? null : (i-1)/2
左孩子:2*i + 1
右孩子:2*i + 2
- 2.堆得定义
n个关键字序列array[0,...,n-1],当且仅当满足下列要求:(0 <= i <= (n-1)/2)
① array[i] <= array[2*i + 1] 且 array[i] <= array[2*i + 2]; 称为小根堆;
② array[i] >= array[2*i + 1] 且 array[i] >= array[2*i + 2]; 称为大根堆;
- 3.建立大顶堆 n个节点的完全二叉树array[0,…,n-1],最后一个节点n-1是第(n-1-1)/2个节点的孩子。对第(n-1-1)/2个节点为根的子树调整,使该子树称为堆。
对于大根堆,调整方法为:若【根节点的关键字】小于【左右子女中关键字较大者】,则交换。
之后向前依次对各节点((n-2)/2 - 1)~ 0为根的子树进行调整,看该节点值是否大于其左右子节点的值,若不是,将左右子节点中较大值与之交换,交换后可能会破坏下一级堆,于是继续采用上述方法构建下一级的堆,直到以该节点为根的子树构成堆为止。
反复利用上述调整堆的方法建堆,直到根节点。
- 4.堆排序(大顶堆)
①将存放在array[0,...,n-1]中的n个元素建成初始堆;
②将堆顶元素与堆底元素进行交换,则序列的最大值即已放到正确的位置;
③将数组中array[0,...,n-1]前n-1个元素再次形成大根堆,再重复第②③步,直到堆中仅剩下一个元素为止。
代码实现
/**
* 大顶堆排序
* @param array
*/
public static void maxHeapSort(int[] array) {
int i;
int len = array.length;
// 构建大顶堆
for (i = len / 2 - 1; i >= 0; i--) {
adjustMaxHeap(array, i, len);
}
// 堆顶是最大值,交换堆顶和最后一个数,再重新调整最大堆,下一次循环 i--
for (i = len - 1; i >= 0; i--) {
int temp = array[0];
array[0] = array[i];
array[i] = temp;
adjustMaxHeap(array, 0, i);
}
System.out.println(Arrays.toString(array));
}
private static void adjustMaxHeap(int[] a, int pos, int len) {
int temp;
int child;
for (temp = a[pos]; 2 * pos + 1 < len; pos = child) {
// 数组从0开始,r(i)>=r(2i) r(i)>=r(2i+1) 对应 pos => 2 * pos + 1 和 2 * pos +2
child = 2 * pos + 1;
// 有右孩子,且右孩子数值更大
if (child + 1 < len && a[child] < a[child + 1]) {
child++;
}
// 最大的孩子大于根节点
if (a[child] > temp) {
a[pos] = a[child];
} else {
break;
}
}
a[pos] = temp;
}
时间复杂度和空间复杂度
时间复杂度:建堆:o(n),每次调整o(log n),故最好、最坏、平均情况下:o(n*logn)。