堆排序
原理:将数组转为完全二叉树再转为大顶堆,然后将根节点与叶子节点的最后一个值调换,找到最大值,将最大值取出,现在又变成一个完全二叉树,继续将这个完全二叉树转成大顶堆,以此类推,直至找到所有的最大值。
升序排序用大顶堆,降序排序用小顶堆
大顶堆:父节点总比两个子节点的权值要大
package com.paixu;
import java.util.Arrays;
public class HeapSort {
public static void main(String[] args) {
int arr[] = new int[] { 9, 6, 8, 7, 1, 0, 10, 4, 2 };
heapSort(arr);
System.out.println(Arrays.toString(arr));// [0, 1, 2, 4, 6, 7, 8, 9, 10]
}
public static void heapSort(int[] arr) {
// 开始位置是最后一个非叶子节点,即最后一个节点的父节点
int start = (arr.length - 1) / 2;
// 调整为大顶堆
for (int i = start; i >= 0; i--) {
maxHeap(arr, arr.length, i);
}
// 把数组中的第0个和堆中的最后一个交换位置,再把前面的处理为大顶堆
for (int i = arr.length - 1; i > 0; i--) {
int temp = arr[0];
arr[0] = arr[i];
arr[i] = temp;
maxHeap(arr, i, 0);
}
}
public static void maxHeap(int[] arr, int size, int index) {
// 左子节点
int leftNode = 2 * index + 1;
// 右子节点
int rightNode = 2 * index + 2;
int max = index;
// 和两个子节点分别对比,找出最大的节点
if (leftNode < size && arr[leftNode] > arr[max]) {
max = leftNode;
}
if (rightNode < size && arr[rightNode] > arr[max]) {
max = rightNode;
}
// 交换位置
if (max != index) {
int temp = arr[index];
arr[index] = arr[max];
arr[max] = temp;
// 交换位置后,可能会破坏之前排好的堆,所以之前排好的堆要重新调整
maxHeap(arr, size, max);
}
}
}