理解堆排序之前,我们首先需要了解顺序存储二叉树的概念:
假设存在一个数组array,我们从下标0开始,按下标顺序将其按照完全二叉树的形式排列,这样构成的二叉树就称为顺序存储二叉树。顺序存储二叉树的节点array[ i ]的左子节点为array[ 2 * i + 1 ],右子节点为array[ 2 * i + 2 ]。
下面进入正题:
堆其实就是具有特殊性质的完全二叉树:若每个节点的值都大于或等于其左右节点的值,则称为大顶堆;若每个节点的值都小于或等于其左右节点的值,则称为小顶堆。
如果用数组来表示一个大顶堆,那么数组应该满足:
array[ i ] >= array[ 2 * i + 1 ] && array[ i ] >= array[ 2 * i + 2 ] (i从0开始)
如果用数组来表示一个小顶堆,那么数组应该满足:
array[ i ] <= array[ 2 * i + 1 ] && array[ i ] <= array[ 2 * i + 2 ] (i从0开始)
一般升序使用大顶堆,降序使用小顶堆。
代码实现:
package DataStructure;
import java.util.Arrays;
public class HeapSort {
public static void main(String[] args) {
int[] arr = {55, 48, 6, 24, 88, 1230, 558};
heapSort(arr);
System.out.println(Arrays.toString(arr));
}
/**
* 将数组调整为大顶堆
* @param arr 需要调整的数组
* @param i 从哪个节点开始
* @param length 需要调整的数组的长度
*/
public static void heapAdjust(int[] arr, int i, int length) {
// 存储当前节点的值
int temp = arr[i];
// 先查找当前节点的左子节点
for (int k = 2 * i + 1; k < length; k = 2 * k + 1) {
// 如果存在右子节点,且比左子节点大,将k替换为右子节点
// k + 1 < length用来判断右子节点是否存在
if (k + 1 < length && arr[k + 1] > arr[k]) {
k++;
}
// 若子节点比当前节点大,将当前节点的值替换为子节点的值
if (arr[k] > temp) {
arr[i] = arr[k];
// 此处先不将temp赋给arr[k],因为之后还有可能继续调整,
// 将k的值赋给i,记录temp需要赋值的位置,
// 若后续不需要调整,则将temp赋值给arr[i],
// 若后续发生了调整,则将temp赋值给子节点的子节点(或者更远)
i = k;
}else {
// 如果temp比左右节点都大了,退出循环
break;
}
}
// 退出循环后,将temp赋值给arr[i]
arr[i] = temp;
}
public static void heapSort(int[] arr) {
// 从序号最大的非叶子节点开始,将数组调整为大顶堆
for (int i = arr.length / 2 - 1; i >= 0; i--) {
heapAdjust(arr, i, arr.length);
}
// 进行堆排序,将堆顶元素与数组末尾元素交换,然后将剩下的元素重新调整为大顶堆
for (int j = arr.length - 1; j > 0; j--) {
int temp = arr[j];
arr[j] = arr[0];
arr[0] = temp;
// 因为只有堆顶元素发生变换,因此从堆顶即0开始调整,
// length设置为j可以将交换后的最大值排除在调整范围外。
heapAdjust(arr, 0, j);
}
}
}