十大排序算法之三 - 堆排序

十大排序算法之三 - 堆排序

1.堆排序介绍

• 堆排序：堆排序（Heapsort）是指利用堆这种数据结构所设计的一种排序算法

• 堆是具有以下性质的完全二叉树

  • 大顶堆：每个结点的值都大于或等于其左右孩子结点的值
    • arr[i] >= arr[2i+1] && arr[i] >= arr[2i+2]
  • 小顶堆：每个结点的值都小于或等于其左右孩子结点的值
    • arr[i] <= arr[2i+1] && arr[i] <= arr[2i+2]

• 堆排序的基本思想是：将待排序序列构造成一个大顶堆，此时，整个序列的最大值就是堆顶的根节点。将其与末尾元素进行交换，此时末尾就为最大值。然后将剩余n-1个元素重新构造成一个堆，这样会得到n个元素的次小值。如此反复执行，便能得到一个有序序列了

2.Java 实现大顶堆（用于 升序排序）

package com.lagou.heapSort;

/**
 * @author 云梦归遥
 * @date 2022/5/18 10:11
 * @description 堆排序 - 升序实现
 */
public class HeapSortMethod {
    // 首先将传入的数组认为是一个完全二叉树，构建大顶堆
    public void createMaxHeap(int[] array, int parentIndex, int length){
        for (int i = parentIndex * 2 + 1; i < length; i = i * 2 + 1){
            // 右孩子存在且右孩子大于左孩子
            if (i + 1 < length && array[i] < array[i + 1]){
                // 右孩子将与父节点进行比较
                i++;
            }
            // 父节点与最大的子节点进行数据交换
            if (array[parentIndex] < array[i]){
                int temp = array[i];
                array[i] = array[parentIndex];
                array[parentIndex] = temp;
                // 继续以子节点为父节点向下遍历（如果第一个父节点不是最后的父节点，则下面是排好序的）
                parentIndex = i;
            } else {
                // 因为下面的节点已经是排好序的了，所以不需要进行判断
                break;
            }
        }
    }

    // 刚传入数组时，构建大顶对；然后循环将堆顶元素交换到数组最后面，数组长度依次减一
    public void heapMaxSort(int[] array){
        // 依次将非叶子节点构建大顶堆（第一个非叶子节点 array.length / 2 - 1 ）
        for (int i = array.length / 2 - 1; i >= 0; i--){
            createMaxHeap(array, i, array.length);
        }

        // 整棵树都是一个大顶堆，然后交换数组中的首尾元素
        int length = array.length;
        int temp = 0;
        while (length > 1){
            temp = array[length - 1];
            array[length - 1] = array[0];
            array[0] = temp;
            // 已经找出最大元素放在数组末尾了，接着先重新构建大顶堆，找次大元素重复操作
            length--;
            // 从根节点开始重构大顶堆
            createMaxHeap(array, 0, length);
        }
    }
}

进行测试

package com.lagou.heapSort.test;

import com.lagou.heapSort.HeapSortMethod;

import java.util.Arrays;

/**
 * @author 云梦归遥
 * @date 2022/5/18 10:36
 * @description 堆排序 - 升序测试
 */
public class HeapSortMethodTest {
    public static void main(String[] args) {
        HeapSortMethod heapSortMethod = new HeapSortMethod();
        int[] array = {3, 2, 5, 7, 4, 1, 9, 6, 8};
        System.out.println("【堆排序 - 前】" + Arrays.toString(array));
        heapSortMethod.heapMaxSort(array);
        System.out.println("【堆排序 - 后】" + Arrays.toString(array));
    }
}

3.总结

• 一般升序采用大顶堆，降序采用小顶堆
• 时间复杂度：O(nlogn)