public class HeapSort {
public static void main(String[] args) {
int[] arr = {4, 6, 8, 5, 9};
heapSort(arr);
System.out.println(Arrays.toString(arr));
}
/**
* 堆排序 大顶堆式
* @param arr 要排序的数组
*/
public static void heapSort(int[] arr) {
//将无序序列构建成一个堆,根据升降序需求选择大顶堆或者小顶堆
//第一次让i等于第一个非叶子节点
for (int i = arr.length / 2 - 1; i >= 0; i--) {
adjustHeap(arr,i,arr.length);
}
//将对顶元素与末尾元素进行交换,将最大元素沉到数组末端
//重新调整结构使其满足堆的定义,然后继续交换堆顶元素与当前末尾元素,反复执行调整+交换步骤。直到序列有序。
for (int j = arr.length - 1; j > 0; j--){
//只需要调数组数量-1个数的位置就可以
//交换
BubbleSort.swap(arr,j,0);
//实际上调整总是从顶上到底下调整,j是顶上,0是底下
adjustHeap(arr,0,j);
}
}
/**
* 将一个数组(二叉树)调整成一个大顶堆'
* 功能;将以i指向的非叶子节点的树调整成大顶堆
* 举例:arr{4,6,8,5,9} i = 1时 经过方法的处理 会得到一个以6为非叶子节点调整过后的数组{4,9,8,5,6}
*
* @param arr 要被调整的数组
* @param i 表示非叶子节点在数组中的位置
* @param length 表示对多少个元素进行调整,length是在逐渐减少
*/
public static void adjustHeap(int[] arr, int i, int length) {
int temp = arr[i]; //保存当前元素(非叶子节点)的值
//k是i节点的左子节点,这个循环就是为了找到最大的左子节点
for (int k = i * 2 + 1; k < length; k = k * 2 + 1) {
if (k + 1 < length && arr[k] < arr[k + 1]) { //比较左子节点的值和右子节点的值
k++; //左子节点更小的话就指向右子节点
}
if (arr[k] > temp){ //如果非叶子节点的右子节点大于非叶子节点,那就互换
arr[i] = arr[k];
i = k; //让i指向k继续循环比较
}else {
//如果左右都没非叶子节点大,那就干脆结束循环
break;
}
}
//当for循环结束后,说明以i为非叶子节点的树,i的位置就是这棵树最大的数了
//将temp值放到调整过后的位置
arr[i] = temp;
}
}