堆排序
堆排序基本介绍
- 堆排序是利用堆这种数据结构而设计的一种排序算法,堆排序是一种选择排序,它的最坏,最好,平均时间复杂度均为O(nlogn),它也是不稳定排序。
- 堆是具有以下性质的完全二叉树:每个结点的值都大于或等于其左右孩子结点的值,称为大顶堆, 注意 : 没有要求结点的左孩子的值和右孩子的值的大小关系。
- 每个结点的值都小于或等于其左右孩子结点的值,称为小顶堆
- 大顶堆举例说明
我们对堆中的结点按层进行编号,映射到数组中就是下面这个样子:
大顶堆特点:arr[i] >= arr[2*i+1] && arr[i] >= arr[2*i+2]
// i 对应第几个节点,i从0开始编号
小顶堆举例说明
小顶堆:arr[i] <= arr[2*i+1] && arr[i] <= arr[2*i+2]
// i 对应第几个节点,i从0开始编号一般升序采用大顶堆,降序采用小顶堆
堆排序基本思想
- 将待排序序列构造成一个大顶堆
- 此时,整个序列的最大值就是堆顶的根节点。
- 将其与末尾元素进行交换,此时末尾就为最大值。
- 然后将剩余n-1个元素重新构造成一个堆,这样会得到n个元素的次小值。如此反复执行,便能得到一个有序序列了。
代码实现
package com.imagpie.tree;
import java.util.Arrays;
import java.util.jar.JarEntry;
public class HeapSort {
public static void main(String[] args) {
//要求将数组进行升序排序
int arr[] = {4, 6, 8, 5, 9};
heapSort(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);
System.out.println(Arrays.toString(arr));
}
for (int i = arr.length-1; i > 0 ; i--) {
temp = arr[i];
arr[i] = arr[0];
arr[0] = temp;
adjustHeap(arr,0,i);
}
System.out.println(Arrays.toString(arr));
}
/**
* 将一个数组(二叉树)调整为大顶堆
*
* @param arr 待调正的数组
* @param i 表示非叶子节点在数组中的索引
* @param length 表示对多少个元素进行调整,length在逐渐减少
*/
public static void adjustHeap(int arr[], int i, int length) {
int temp = arr[i];//取出当前元素的值,保存在临时变量中
/**
* 开始调整
* 说明
* 1.k = i*2+1表示的的是i的左子结点
*/
for (int k = i * 2 + 1; k < length; k = k * 2 + 1) {
if (k + 1 < length && arr[k] < arr[k + 1]) {
//说明左子结点的值小于右子结点的值
k++;
}
if (arr[k] > temp) {
//如果子结点的值大于父节点的值,就进行交换
arr[i] = arr[k];
i = k;
} else {
break;//因为是从最后一个非叶子节点开始的,所以当如果子结点小于父节点的话直接可以退出循环
}
}
//当for循环结束后,我们已经将以i为父节点的树的最大值,放在了最顶端
arr[i] = temp;
}
}