堆排序

前言

最近在学数据结构与算法，在此做一下记录，方便时不时复习一下。如果有什么做得不对的地方，恳请指出，万分感谢。

简述

应用场景：许多应用程序都需要处理有序的元素，但不一定要求它们全部有序，或者不一定要一次就将它们排序，很多时候每次只需要操作数据中的最大(最小)元素，那么有一种基于二叉堆的数据结构可以提供支持。

二叉堆：就是一个完全二叉树的结构，同时满足堆的性质：即子节点的键值或索引总小于(或者大于)它的父节点。在一个二叉堆中，根节点总是最大(或者最小)节点。

堆排序算法就是抓住了这一特点，每次都取堆顶的元素，然后将剩余的元素重新调整为最大(最小)堆，依次类推，最终得到排序的序列。

补充知识点

需要理解什么叫二叉树、满二叉树、完全二叉树的概念，以及图中的两个推论，推论1可以用于访问当前节点的左右节点(如果有的话)，推论2可以用于构建最大堆时的起始位置。

步骤

第一步：将该数组看成一个完全二叉堆；

第二步：找到最后一个非叶子节点的位置，然后将它和自己的节点进行比较和交换；

第三步：继续按照从下到上，从右往左调整每一个非叶子节点，调整完后即初始化完成，最大(小)的数此时就成为了根节点。

第四步：开始排序，将堆顶的元素和尾元素进行交换，此时最大(最小)的数就已经被取出；

第五步：交换完之后，重新调整成为二叉堆，此时调整的长度为上次调整的长度减一，调整完后，继续将堆顶的元素和尾元素进行交换；

第六步：重复第五步，调整数组长度只要大于0，就一直重复第五步，直至调整完，此时数组已经排序完成。

图示

 

代码

public class HeapSort {

    private static int len ;//用来计算数组长度

    public final static int[] DATA = {32,61,44,8,84,22,51,10};

    public static void main(String[] args) {
        printArray(DATA);
        System.out.println("=======================");
        printArray(sort(DATA));
    }


    public static int[] sort(int[] arrays) {
        len = arrays.length;
        if (len < 1) return arrays;
        //构建一个最大堆
        buildMaxHeap(arrays);
        //取出堆顶元素和尾元素交换
        while (len > 0) {
            swap(arrays,0,len-1);
            len--;
            adjustHeap(arrays,0);
        }

        return arrays;
    }

    /**
     * 构建一个最大堆
     * @param arrays 数组
     */
    private static void buildMaxHeap(int[] arrays) {
        for (int i = (len/2 - 1); i >= 0; i--) {
            adjustHeap(arrays,i);
        }
    }

    /**
     * 调整成为新的堆
     * @param arrays 数组
     * @param i 根节点
     */
    private static void adjustHeap(int[] arrays, int i) {
        int maxIndex = i;//保存最大元素的索引
        int left = 2*i+1;//左子节点
        int right = 2*(i+1);//右子节点
        if (left < len && arrays[left] > arrays[maxIndex]) {//这种状态下 <:降序 >:升序
            maxIndex = left;
        }
        if (right < len && arrays[right] > arrays[maxIndex]) {//这种状态下 <:降序 >:升序
            maxIndex = right;
        }
        if (maxIndex != i) {
            swap(arrays,maxIndex,i);
            adjustHeap(arrays,maxIndex);//重新调整为二叉堆
        }

    }

    /**
     * 用于交换数组内两个元素
     * @param arrays
     * @param i
     * @param j
     */
    public static void swap(int[] arrays,int i,int j) {
        int temp = arrays[i];
        arrays[i] = arrays[j];
        arrays[j] = temp;
    }

    public static void printArray(int[] array) {
        for (int i:array) {
            System.out.print(i + " ");
        }
        System.out.println("");
    }
}

结果