数据结构(Java)学习笔记——知识点:堆和堆排序
十四、堆和堆排序
1、基本概念
堆:是具有以下性质的完全二叉树:每个节点的值都大于或等于其左右孩子节点的值,称为大顶堆。不要求左右孩子的大小关系。相当于满足 arr[i] > arr[ 2 * i + 1] && arr[i] > arr[ 2 * i + 2]
相反,每个节点的值都小于或等于其左右孩子节点的值,称为小顶堆。arr[i] < arr[ 2 * i + 1] && arr[i] < arr[ 2 * i + 2]
堆排序:堆排序是利用堆这种数据结构而设计的一种排序算法,堆排序是一种选择排序,它的平均时间复杂度为O(nlogn),它也是不稳定排序。
一般升序排序采用大顶堆,降序采用小顶堆。
2、堆排序思想
需求:给一个数组{4,6,8,5,9},要求使用堆排序的方法,将数组升序排序。
1)将待排序序列构造成一个大顶堆(数组)
2)整个序列的最大值就是堆顶的根节点
3)将其与末尾元素进行交换,此时末尾就是最大值了
4)然后将剩余 n - 1 个元素重新构造成一个堆,这样会得到 n 个元素的次小值。如此反复,就得到了一个有序序列。
现在我们来细说:
1)将待排序序列构造成一个大顶堆(数组)
第一步:假设给定无序序列结构如下。
第二步:判断最后一个非叶子节点是否它的根节点有比它的像个孩子要小,如果有,则交换他们的位置(6和9互换)。
第三步:向上查找,直到根节点,发现 [4,8,9] 中的 9是最大的,将 4 和 9 交换。
第四步:现在[ 4,5,6 ]的结构混乱了,继续调整,这里面 6 最大,交换 4 和 6 的位置。
此时,一个无序序列就构造成了一个大顶堆。
2)整个序列的最大值就是堆顶的根节点
3)将其与末尾元素进行交换,此时末尾就是最大值了
第一步:堆顶 9 和 末尾 4 交换。
第二步:重新调整结构(8 和 4 互换),使其继续满足堆定义。
第三步:堆顶 8 和 5 进行交换,得到第二大元素 8。
4)然后将剩余 n - 1 个元素重新构造成一个堆,这样会得到 n 个元素的次小值。如此反复,就得到了一个有序序列。
最后,变成了:
3、堆排序代码实现
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;//中间值,用作交换使用
System.out.println("堆排序:");
//分布完成 一步完成
/*
adjustHeap(arr,1,arr.length);
System.out.println("第1次" + Arrays.toString(arr));//4,9,8,5,6
adjustHeap(arr,0,arr.length);
System.out.println("第2次" + Arrays.toString(arr));//9,6,8,5,4
*/
//完成我们最终代码
//将无序序列构建成一个堆,根据升序降序需求选择大顶堆或小顶堆
for(int i = arr.length / 2 - 1; i >= 0; i--){//这样就是倒序的构建堆,从下往上
adjustHeap(arr, i, arr.length);
}
/*
3)**将其与末尾元素进行交换**,此时末尾就是最大值了
4)然后将剩余 n - 1 个元素重新构造成一个堆,这样会得到 n 个元素的次小值。**如此反复**,就得到了一个有序序列。
*/
for(int j = arr.length - 1; j > 0; j--){
//交换,把最大的那个只逐个放到第一个位置,倒着放
temp = arr[j];
arr[j] = arr[0];
arr[0] = temp;
adjustHeap(arr, 0, j);//重新构建成堆
}
System.out.println(Arrays.toString(arr));
}
/**
* @Author: Cui
* @Description: 将一个数组调整成大顶堆的样子
* @DateTime: 18:45
* @Params: arr:要调整的数组,i:表示非叶子节点在数组中的索引,length:表示数组长度
* @Return
*/
public static void adjustHeap(int[] arr, int i, int length){
int temp = arr[i];//先取出当前元素的值,保存在临时变量中。
for(int k = i * 2 + 1; k < length; k = k * 2 + 1){
if(k + 1 < length && arr[k] < arr[k + 1]){//如果当前元素的左子节点,小于右子节点
k++;//k 指向右子节点
}
if(arr[k] > temp){//如果子节点大于父节点,那么需要交换两个树
arr[i] = arr[k];//把较大的值赋值给当前节点
i = k;//!!!,这个很重要,i 指向 k ,继续循环比较。
}else{
break;
}
}
//当for循环结束后,我们已经将i为父节点的树的最大值,放在了最顶端。
//将temp值放在调整之后的位置
arr[i] = temp;
}
}
运行结果
堆排序:
[4, 5, 6, 8, 9]
4、堆排序处理10w条数据的速度
排序前的时间:2023-07-27 20:28:10:455
排序后的时间:2023-07-27 20:28:10:483
可以看到堆排序处理10w条数据的速度是28ms,当然,数据越多,越能看出数据处理的高效性。