堆是具有以下性质的完全二叉树:
每个结点大于等于它的左右孩子结点的值称为最大堆或大顶堆;
每个结点小于等于它的左右孩子结点的值称为最小堆或小顶堆;
1.具体实现
public class HeapSort {
public static void main(String[] args){
int[] a = {3,4,6,1,2,9,8,0,5,7};
System.out.println(Arrays.toString(sort(a)));
}
/**
* 顺序: 从下往上,从右到左
* @param arr
* @return
*/
private static int[] sort(int[] arr){
/**
* 构建最大堆
*/
for(int i = arr.length/2-1; i >= 0; i--){ //数组从0开始 10个元素 最后一个非叶子结点应该是4
heapSort(arr, i, arr.length);
}
/**
* 调整堆,每一次都将最大值放置末尾
*/
for(int i = arr.length-1; i > 0; i--){
swap(arr, 0, i);
heapSort(arr, 0, i-1);
}
return arr;
}
private static void swap(int[] arr, int index1, int index2){ //用于交换根节点与最后一个结点
int temp = arr[index1];
arr[index1] = arr[index2];
arr[index2] = temp;
}
/**
* 从最后一个非叶子结点开始 很关键
* 位置图
* 0
* 1 2
* 3 4 5 6
* 7 8 9
* @param arr
* @param index
* @param length
*/
private static void heapSort(int[] arr, int index, int length){
int temp = arr[index];
for(int i = index*2 + 1; i < length; i=i*2+1){ //循环用于调整堆, 构建堆可用if
//+1 很关键 可以画图
if(i+1 < length && arr[i] < arr[i+1]){
++i;
}
if(temp > arr[i]){
break;
}
arr[index] = arr[i];
index = i;
}
arr[index] = temp;
}
}
2.测试结果
[0, 2, 1, 3, 4, 5, 6, 7, 8, 9]