public class HeapSort {
public void sort(int arr[]) {
int n = arr.length;
// 构建最大堆
for (int i = n / 2 - 1; i >= 0; i--)
heapify(arr, n, i);
// 从最后一个非叶子节点开始进行堆排序
for (int i = n - 1; i > 0; i--) {
// 将堆顶元素(最大值)与当前未排序部分的最后一个元素交换
int temp = arr[0];
arr[0] = arr[i];
arr[i] = temp;
// 重新构建最大堆
heapify(arr, i, 0);
}
}
// 构建最大堆
void heapify(int arr[], int n, int i) {
int largest = i; // 初始化最大值索引
int l = 2 * i + 1; // 左子节点索引
int r = 2 * i + 2; // 右子节点索引
// 如果左子节点比父节点大,则更新最大值索引
if (l < n && arr[l] > arr[largest])
largest = l;
// 如果右子节点比父节点大,则更新最大值索引
if (r < n && arr[r] > arr[largest])
largest = r;
// 如果最大值索引不是当前节点,则交换当前节点与最大值节点,并继续向下调整堆
if (largest != i) {
int swap = arr[i];
arr[i] = arr[largest];
arr[largest] = swap;
heapify(arr, n, largest);
}
}
public static void main(String args[]) {
int arr[] = {12, 11, 13, 5, 6, 7};
HeapSort ob = new HeapSort();
ob.sort(arr);
System.out.println("排序后的数组:");
for (int i = 0; i < arr.length; i++)
System.out.print(arr[i] + " ");
}
}
Java中堆排序
最新推荐文章于 2024-05-20 18:16:07 发布