import java.util.Arrays;
/**
* 堆排序
*
*/
public class HeapSort {
/**
* 调整大顶堆
* @param arr 待排序列
* @param parent 父节点
* @param len 待排序列尾元素索引
*/
public static void adjustHeap(int[] arr, int i, int len) {
int temp = arr[i]; //先取出i位置的元素
for(int k = i*2 + 1; k < len; k = k*2 + 1) { //从i的左节点开始
if(k+1 < len && arr[k] < arr[k+1]) { //左节点小于右节点,k指向右节点
k++;
}
if(arr[k] > temp) { //如果子节点大于父节点,将子节点赋给父节点
arr[i] = arr[k];
i = k; //向下调整
}
else {
break;
}
}
arr[i] = temp; //将temp值放到最终位置上
}
/**
* 堆排序
* @param arr
*/
public static void HeapSort(int[] arr) {
int i;
//1.构建大顶堆
for(i = arr.length / 2 - 1; i >= 0; i--) {
//从第一个非叶子节点从下往上,从右往左调整结构
adjustHeap(arr, i, arr.length);
}
//2.调整堆结构+交换堆顶和末尾元素
for(int j = arr.length - 1; j > 0; j--) {
swap(arr, 0, j); //交换堆顶和末尾元素
adjustHeap(arr, 0, j); //重新调整堆
}
}
/**
* 交换元素
* @param arr
* @param i
* @param j
*/
public static void swap(int[] arr, int i, int j) {
int temp = arr[i];
arr[i] = arr[j];
arr[j] = temp;
}
public static void main(String[] args) {
int[] arr = {2, 9, 6, 4, 3, 8, 7, 11, 1};
HeapSort(arr);
System.out.println(Arrays.toString(arr));
}
}
运行结果
参考
https://www.cnblogs.com/chengxiao/p/6129630.html