学习堆排序必须搞明白堆这个数据结构!!!
排序思路:
首先将待排序数组构建成最大堆,然后逐一将root放到末尾,堆的特性会将其第二大的数作为最大数上浮成root,最终产生有序数组。
代码:
import java.util.Arrays;
/**
* 堆排序
*
* @author DepengZhang
* @date 2021年07月30日 20:01
*/
public class HeapSort {
/**
* 下沉调整
*
* @param array 待调整的堆
* @param parentIndex 要下沉的父节点
* @param length 堆的有效大小
*/
public static void downAdjust(int[] array, int parentIndex, int length) {
// temp 保存父节点值,用于最后的赋值
int temp = array[parentIndex];
int childIndex = 2 * parentIndex + 1;
while (childIndex < length) {
// 如果有右孩子,且右孩子大于左孩子的值,则定位到右孩子
if (childIndex + 1 < length && array[childIndex + 1] > array[childIndex]) {
childIndex++;
}
// 如果父节点大于任何一个孩子的值,则直接跳出
if (temp >= array[childIndex]) {
break;
}
//无须真正交换,单向赋值即可
array[parentIndex] = array[childIndex];
parentIndex = childIndex;
childIndex = 2 * childIndex + 1;
}
array[parentIndex] = temp;
}
/**
* 堆排序(升序)
*
* @param array 待调整的堆
*/
public static void heapSort(int[] array) {
//把无序数组构建成最大堆
for (int i = (array.length - 2) / 2; i >= 0; i--) {
downAdjust(array, i, array.length);
}
System.out.println(Arrays.toString(array));
//循环删除堆顶元素,移到集合尾部,调整堆产生新的堆顶
for (int i = array.length - 1; i > 0; i--) {
// 最后1个元素和第1个元素进行交换,这个交换建议还是换成temp转,一旦数值一样此两个值将变成0
array[i] = array[i] ^ array[0];
array[0] = array[i] ^ array[0];
array[i] = array[i] ^ array[0];
// “下沉”调整最大堆
downAdjust(array, 0, i);
}
}
public static void main(String[] args) {
int[] arr = new int[]{1, 3, 2, 6, 5, 7, 8, 9, 10, 0};
heapSort(arr);
System.out.println(Arrays.toString(arr));
}
}