利用堆的特性
大顶堆–升序排序
1.将数组看作顺序存储的完全二叉树
2.将完全二叉树–>大顶堆
3.将堆顶元素交换至数组末尾,作冒泡处理
4.对剩下部分大顶堆化
import java.util.Arrays;
public class HeapSort {
/**
* 将index按照大顶堆的规则进行排序,适用于最后的父节点\其后有序的大顶堆节点
* @param arr 进行排序的顺序存储的二叉树(堆)
* @param size 想要排序的范围,用于冒泡.在后面排序的时后遍历的size需要逐渐减少
* @param index 需要向下进行maxHeap的节点,从这个节点开始工作
*/
public static void maxHeap(int[] arr,int size,int index) {
int left = index*2+1; //取出index的左右儿子
int right = index*2+2;
int max = index;
if(left<size && arr[left]>arr[max]) {//一是防止越界,二是在需要的范围内进行maxHeap
max = left;
}
if(right<size && arr[right]>arr[max]) {
max = right;
}
if(max!=index) { //如果最大值就在index处,递归结束
int temp = arr[index];
arr[index] = arr[max];
arr[max] = temp;
maxHeap(arr,size,max); //交换后的子节点可能会比它的子节点大
//所以要在size范围内向后给新调换的子节点进行maxHeap
}
}
public static void heapSort(int[] arr) {
int index = (arr.length-1-1)/2; //取最后节点的父节点
for(int i=index;i>=0;i--) { //从该点向前遍历并maxHeap,用来制造出大顶堆
maxHeap(arr,arr.length-1,i);//因为是第一遍整理,这里size必须包括堆的全部
}
for(int i=arr.length-1;i>0;i--) { //这里开始排序,将冒泡到堆顶的元素换到末尾,
int temp = arr[0]; //并且在下次maxHeap时不再取到
arr[0] = arr[i];
arr[i] = temp;
/*
* 交换后,除了最大值外,继续对剩下的堆进行大顶堆化
* 此处size必须逐渐减小,保证冒到最后的元素不受maxHeap的影响
* 此时需要向下进行maxHeap的元素只有这一个刚换到顶端的元素
*/
maxHeap(arr,i,0);
}
}
public static void main(String[] args) {
int[] arr = new int[] {9,6,8,7,0,1,10,4,2};
heapSort(arr);
System.out.println(Arrays.toString(arr));
}
}