堆排序
时间复杂度O(nlogn),不稳定排序。
堆是具有一下性质的完全二叉树:每个节点的值都大于或等于其左右孩子节点的值,称为大顶堆;每个节点的值都小于或等于左右孩子节点的值,称为小顶堆。(不要求左右孩子节点的大小关系)
大顶堆特点:arr[i] >= arr[2i+1] && arr[i] >= arr[2i+2]
小顶堆特点:arr[i] <= arr[2i+1] && arr[i] <= arr[2i+2]
升序使用大顶堆,降序使用小顶堆
i是从0开始编号的:
基本思想:
- 将无序序列构建成一个堆,根据升序降序需求选择大顶堆或小顶堆
- 将堆顶元素与末尾元素交换,将最大元素沉到数组末端
- 将剩余n-1个元素重新构造成一个堆,会得到n个元素的次小值,如此反复执行,能得到一个有序序列
代码思想:
- 先写一个函数adjust:将一个数组(二叉树)调整成一个大顶堆
- 参数为 1⃣️数组,2⃣️非叶子节点在数组中的索引i,3⃣️表示对多少个元素进行调整
- 首先将当前非叶子节点的值存入temp
- 写一个for循环,k设置为左子节点,k<length时停止循环,每次循环k=k的左子节点
- 查看当前非叶子节点左右子节点哪个大(且满足k+1<num,防止已经放到结尾的数字再被拿出来),取较大值的索引为k
- 若子节点大于父节点,把较大的值赋值给函数的索引节点i,将k赋值给i,继续循环比较;否则,break当前for循环。
- 循环结束,将temp的值赋值给arr[i],表示将temp值放到调整后的位置。
- 创建for循环,循环调用adjust,
for(int i = arr.length/2-1; i>=0; i--)
,for循环完成时,大顶堆就构造完成了 - 创建for循环,将堆顶元素与末尾元素进行交换,将最大的元素沉到数组末端;将剩余元素进行adjust,
代码:
package com.sortAlgorithm;
import java.util.Arrays;
public class heapSort {
public static void main(String[] args){
int[] arr = {4,6,8,5,9};
for(int i = arr.length/2-1; i>=0; i--){
adjust(arr,i,arr.length);
}
for(int j = arr.length-1; j>0; j--){
int temp = arr[j];
arr[j] = arr[0];
arr[0] = temp;
adjust(arr,0,j);
}
System.out.println(Arrays.toString(arr));
}
/**
* 构建大顶堆的函数
* @param arr 传入的数组
* @param i 非叶子节点在数组中的索引
* @param num 对多少个元素进行调整
*/
public static void adjust(int[] arr, int i, int num){
int temp = arr[i];
for(int k = i*2+1; k< num; k = k*2+1){
if(arr[k]<arr[k+1]&&k+1<num){
k++;
}
if(arr[k]>temp){
arr[i] = arr[k];
i = k;
}else{
break;
}
}
arr[i] = temp;
}
}