排序算法——堆排序

堆的特点

1、 是一棵完全二叉树
2、 所有父节点的值大于子结点的值,或者所有父节点的值小于子结点的值

完全二叉树的特点

1、是一棵二叉树
2、生成结点顺序:从上到下,从左往右,填满一层再填下一层

基于完全二叉树的特点,堆可以用一维数组来存储

设某个结点在数组中的下标为 i ,则:

  • 其父结点在数组的下标为 (i - 1) / 2 ,向下取整
  • 其左子结点在数组的下标为 2 * i + 1
  • 其右子结点在数组的下标为 2 * i + 2

大顶堆和小顶堆

  • 任意结点的值都大于其子结点的值——大顶堆

  • 任意结点的值都小于其子结点的值——小顶堆

大顶堆得到的是升序排序

小顶堆得到的是降序排序

因为每次拿的都是根结点,然后放在数组最后面

堆排序算法

1、堆化,反向调整使得每个子树都是大顶堆或者小顶堆

2、按序输出元素:把顶堆和最末元素对调,然后调整顶堆元素

思路:大顶堆初始化之后,根结点是最大值,跟最后一个元素交换,实际上就是把最大值放到数组最后面,然后再进行堆调整,又可以得到一个最大值,不断重复,就得到了一个有序的数组了。

代码

public class Main {

    public static void main(String[] args) {
        int[] arr = new int[]{1, 2, 3, 4, 5, 6};

        sort(arr);
        
        for (int i : arr) {
            System.out.println(i);
        }
    }
    
    /**
     * 堆排序
     * @param arr   完全二叉树的数组存储
     */
    public static void sort(int[] arr) {
        int n = arr.length;
        buildHeap(arr, n);

        for (int i = n - 1; i >= 0; i--) {
            // 把堆顶,0号元素和最后一个元素对调
            swap(arr, 0, i);
            // 缩小堆的范围,对堆顶元素进行向下调整
            heapfiy(arr, i, 0);
        }

    }

    /**
     * 初始化大顶堆,从下至上,调整堆
     * @param arr   完全二叉树的数组存储
     * @param n     结点个数 = 数组的长度
     */
    public static void buildHeap(int[] arr, int n) {
        int last = n - 1;
        int parent = (last - 1) / 2;

        for (int i = parent; i >= 0 ; i--) {
            heapfiy(arr, n, i);
        }


    }


    /**
     * 调整堆
     * @param arr   完全二叉树的数组存储
     * @param n     结点个数 = 数组的长度
     * @param i     结点在数组中的下标
     */
    public static void heapfiy(int[] arr, int n, int i) {

        if (i >= n) {
            return;
        }

        int left = 2 * i + 1;
        int right = 2 * i + 2;
        int max = i;

        if (left < n && arr[left] > arr[max]) {
            max = left;
        }

        if (right < n && arr[right] > arr[max]) {
            max = right;
        }

        if (max != i) {
            swap(arr, max, i);
            heapfiy(arr, n, max);
        }
    }

    /**
     * 交换两个值
     */
    public static void swap(int[] arr, int x, int y) {
        int temp = arr[x];
        arr[x] = arr[y];
        arr[y] = temp;
    }
}

你可能感兴趣的:(排序算法——堆排序)