https://www.jianshu.com/p/11655047ab58
参考以上链接理解堆排序。
堆实际上是一种特殊的完全二叉树,从根开始,其子树的根结点大于其左右子节点的值,递归下去依然如此,即这是大根堆。相反,其子树的根节点小于其左右子节点的值,递归下去依然如此,则这是小根堆。
依据堆的定义可知,小根堆中的根节点为堆中的最小值,大根堆中的根结点为堆中的最大值,因此堆排序,就是可将大根堆或者小根堆的根节点与堆的最后一个元素交换,再调整除最后一个元素的堆结构,调整后堆的根为整个未排序堆中的最大值(最小值),将其与未排序的堆的最后一个元素交换,以此下去,堆排序即可完成。
所以,堆排序可分为两步:
1.初始化堆
2.交换堆的根结点与最后一个元素
3.交换完毕之后调整堆结构
package bs;
import java.util.Scanner;
public class HeapSort {
//这个函数只是从当前结点向下调整堆结构,
public static void adjustHeap(int[] arr,int pos, int end ){
int temp = arr[pos];
int k = 2*pos+1;
//从当前位置的子节点开始调整
for(;k<=end;k = 2*k+1){
if(k+1<=end&&arr[k]<arr[k+1]){
k++;
}
if(arr[k]>temp){
arr[pos] = arr[k];
arr[k] = temp;
}
else {
break;
}
pos = k;
}
}
public static void swap(int[] arr, int i, int j){
arr[i] = arr[i] ^ arr[j];
arr[j] = arr[i] ^ arr[j];
arr[i] = arr[i] ^ arr[j];
}
public static void sort(int[] arr){
if(arr==null||arr.length<=0){
return;
}
int len = arr.length;
//初始化堆,这里的循环体现从右向左,从上到下调整堆结构,而里面的函数只是从上到下调整堆。
for(int i=len/2-1;i>-1;i--){
adjustHeap(arr, i, len-1);
}
//交换堆根与末尾元素,交换完毕之后再从上到下调整堆
for(int i=len-1;i>0;i--){
swap(arr, 0, i);
adjustHeap(arr, 0, i-1);
}
}
public static void main(String[] args){
Scanner scanner = new Scanner(System.in);
int len = scanner.nextInt();
scanner.nextLine();
int[] arr = new int[len];
for(int i=0;i<len;i++){
arr[i] = scanner.nextInt();
}
//排序
sort(arr);
for(int i=0;i<len;i++){
System.out.print(arr[i]+" ");
}
}
}