数据结构与算法——堆排序

1. 基本介绍

1）堆排序是利用堆这种数据结构而设计的一种排序算法，对排序是一种选择排序，它的最坏，最好，平均时间复杂度均为O(nlogn)，它也是不稳定排序。
2）堆是具有以下性质的完全二叉树：每个节点的值都大于等于其左右孩子节点的值，称为大顶堆，注意：没有要求节点的左孩子节点大于右孩子节点的值。
3) 每个节点的值都小于或等于其左右孩子节点的值。称为小顶堆。
大顶堆特点：arr[i] >= arr[i * 2 + 1] && arr[i] >= arr[i*2 + 2]// i 对应第几个节点，i从0开始编号
小顶堆特点：arr[i] <= arr[i * 2 + 1] && arr[i] <= arr[i*2 + 2]// i 对应第几个节点，i从0开始编号

2. 基本思路

1）将无序序列构建成一个堆，根据升降序需求选择大顶堆或小顶堆
2）将堆顶元素与末尾元素交换，将最大元素“沉”到数组末端
3）重新调整结构，使其满足堆定义，然后继续交换堆顶元素与当前末尾元素，反复执行调整+交换步骤，直到整个序列有序。

import java.lang.reflect.Array;
import java.util.Arrays;

public class HeapSort {
    public static void main(String[] args) {
        // 将需要排序数组升序排列
        int[] arr = {4,6,8,5,9};


    }

    // 编写一个堆排序方法
    public static void heapSort(int[] arr) {
        System.out.println("堆排序！");

        int temp = 0;
        // 将无序序列构建成一个堆，根据升降序需求选择大顶堆或小顶堆
        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);
        }
        System.out.println("数组=" + Arrays.toString(arr));
    }

    // 将一个数组（二叉树），调整成一个大顶堆

    /**
     * 功能：完成将以i对应的非叶子节点的树调整成大顶堆
     * @param arr 待调整的数组
     * @param i 表示非叶子节点在数组中的索引
     * @param length 表示对多少个元素继续调整， length是在逐渐减少的
     */
    public static void adjustHeap(int[] arr, int i, int length) {
        int temp = arr[i];// 先取出当前元素的值，保存在临时变量中
        // 开始调整
        // 说明
        // 1、k = i * 2 + 1 是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;// !!! i 指向 k，继续循环比较
            } else {
                break;
            }
        }
        // 当for循环结束后，我们已经将以 i为父节点的树的最大值放在了最顶（局部）
        arr[i] = temp;// 将temp放到调整后的位置
    }

}

堆排序的速度非常快，在我的机器上8百万数据3秒左右。O(nlogn)