- 堆排序是利用堆这种数据结构而设计的一种排序算法,堆排序是一种选择排序,它的最坏,最好,平均时间复 杂度均为 O(nlogn),它也是不稳定排序。
- 将待排序序列构造成一个大顶堆
- 此时,整个序列的最大值就是堆顶的根节点。
- 将其与末尾元素进行交换,此时末尾就为最大值。
- 然后将剩余 n-1 个元素重新构造成一个堆,这样会得到 n 个元素的次小值。如此反复执行,便能得到一个有序 序列了。
/**
* @Description: 堆排序
* @Date: 2021/7/26 18:27
*/
public class HeapSort {
public static void main(String[] args) {
int[] arr = {5, 4, 6, 9, 8};
heapSort(arr);
System.out.println(Arrays.toString(arr));
}
public static void heapSort(int[] arr) {
int temp = 0;
// 将整个数组调整成大顶堆
for (int i=arr.length/2 - 1; i>=0; i--) {
adjustHeap(arr, i, arr.length);
}
// 将堆顶元素与最后一个元素交换,让最大的数沉到数组末尾
for (int j=arr.length-1; j>0; j--) {
temp = arr[j];
arr[j] = arr[0];
arr[0] = temp;
// 这里是为了将第一个叶子节点和它的两个子树进行调整,确保根节点是最大的数
adjustHeap(arr,0,j);
}
}
/**
* 最后一个非叶子节点所在的子树调整成大顶堆
* @param arr 待调整的数组
* @param i 非叶子节点在数组中的位置
* @param length 待排数组的长度,这个值是逐渐递减的
*/
public static void adjustHeap(int[] arr,int i, int length) {
// 取出当前待排非叶子节点的值
int temp = arr[i];
// 开始调整,k是i的左子节点
for (int k = i*2 + 1; k < length; k = k*2 + 1) {
// 将左子节点和右子节点进行比较
if (arr[k] < arr[k+1] && k+1 < length) {
k++;
}
// 将两个子节点的较大值与非叶子结点i进行比较
if (arr[k] > temp) {
// 把较大的赋值给i
arr[i] = arr[k];
// 同时将子节点的位置保留
i = k;
}else {
// 如果非叶子节点最大,直接退出for循环
break;
}
}
// 如果非叶子节点本来就是最大的,这里进行交换就不改变i的值
// 如果进行了交换,这里就是将较小的非叶子节点换到叶子节点k上,上面做交换的时候,将k的值赋值给了i
arr[i] = temp;
}
}