public static void main(String[] args) {
int[] arr = {12, 11, 13, 5, 6, 7};
heapSort(arr);
for (int value : arr) {
System.out.print(value + " ");
}
}
// 堆排序的主要函数
public static void heapSort(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);
}
}
// 将以节点 i 为根的子树调整为最大堆
public static void heapify(int[] arr, int n, int i) {
int largest = i;
int left = 2 * i + 1;
int right = 2 * i + 2;
// 如果左子节点大于根节点,将 largest 设为左子节点的索引
if (left < n && arr[left] > arr[largest]) {
largest = left;
}
// 如果右子节点大于根节点,将 largest 设为右子节点的索引
if (right < n && arr[right] > arr[largest]) {
largest = right;
}
// 如果最大值不是根节点,交换根节点和最大值节点
if (largest != i) {
int swap = arr[i];
arr[i] = arr[largest];
arr[largest] = swap;
// 递归调整子树
heapify(arr, n, largest);
}
}
堆排序-Java
最新推荐文章于 2024-11-11 21:19:25 发布