编辑代码

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("Sorted array is");
        for (int i : arr) {
            System.out.print(i + " ");
        }
    }
}