java实现堆排序

java实现堆排序

/**
 * 推排序
 */
public static void heapSort(int[] arr) {
    int temp;
    // 生成大顶堆
    for (int i = arr.length / 2 - 1; i >= 0; i--) {
        adjustHeap(arr, i, arr.length);
    }
    // 栈顶和末尾元素交换
    for (int j = arr.length - 1; j > 0; --j) {
        // 交换
        temp = arr[j];
        arr[j] = arr[0];
        arr[0] = temp;
        adjustHeap(arr, 0, j);
    }
}

/**
 * 调整为一个大顶堆
 *
 * @param arr    待调整的数组
 * @param i      非叶子节点在数组中索引
 * @param length 对多少元素继续调整
 */
private static void adjustHeap(int[] arr, int i, int length) {
    // 取出当前非叶子节点
    int temp = arr[i];
    // 调整
    for (int k = i * 2 + 1; k < length; k = k * 2 + 1) {
        // 左右子节点相比
        if (k + 1 < length && arr[k] < arr[k + 1]) {
            ++k;
        }
        // 大的子节点和该节点相比
        if (arr[k] > temp) {
            arr[i] = arr[k];
            i = k;
        } else {
            break;
        }
    }
    arr[i] = temp;
}