堆排序是一种基于堆数据结构的排序算法。堆可以看作是一棵完全二叉树,它分为最大堆和最小堆两种。最大堆指父节点的值大于子节点的值,最小堆则相反。
堆排序的基本思想是:将待排序的序列构成一个堆,堆顶元素为最大(或最小)值,将堆顶元素与堆底元素交换,然后将剩余的元素重新构成堆,重复执行交换和重构堆的操作,直到整个序列有序。
以下是 Java 实现堆排序的代码:
class Main {
public static void main(String[] args) {
// 示例
int[] arr = {12, 2, 5, 9, 3};
sort(arr);
for(int i=0; i<arr.length; i++) {
System.out.print(arr[i] + " ");
}
}
//堆排序
public static void sort(int[] arr) {
int len = arr.length;
for (int i = len / 2 - 1; i >= 0; i--) {
heapify(arr, len, i);
}
for (int i = len - 1; i >= 0; i--) {
swap(arr, 0, i);
heapify(arr, i, 0);
}
}
//heapify()构造最大堆
private static void heapify(int[] arr, int len, int i) {
int largest = i;
int l = 2 * i + 1; //左结点
int r = 2 * i + 2; //右节点
if (l < len && arr[l] > arr[largest]) {
largest = l;
}
if (r < len && arr[r] > arr[largest]) {
largest = r;
}
if (largest != i) {
swap(arr, i, largest);
heapify(arr, len, largest);
}
}
//swap()交换数组中的两个元素
private static void swap(int[] arr, int i, int j) {
int temp = arr[i];
arr[i] = arr[j];
arr[j] = temp;
}
}
运行结果如下:
堆排序算法的时间复杂度为 O(nlogn)。