堆排序

堆排序

前言

如果对堆这个数据结构不了解，可以参照我的另一篇博客优先队列

例子

我们要对3，5，7，1，2，9进行顺序排序。利用堆排序怎么处理，如下图所示

最大堆调整

可以看出来这个序列并不符合最大堆的性质，我们需要对这个堆进行调整，从最后一个非叶子节点7开始调整
第一步调整7，7<9,所以交换即可得下图

第二步对5进行调整，发现5>2,无需调整
第三步对3进行调整3<9,交换3和9，3<7,交换3和7得下图，此时已经符合最大堆性质

如何排序

最大堆堆顶元素最大，可每次把堆顶元素与末尾元素交换，堆尾元素隐式出堆，然后再对堆顶进行一次调整算法。下面来演示这个过程

第一步9与3交换，堆尾隐式出堆

对堆顶进行一次调整算法

第二步7与2交换，堆尾隐式出堆

对堆顶进行一次调整算法

第三步5与1交换，堆尾隐式出堆

对堆顶进行一次调整算法

第四步3与1交换，堆尾隐式出堆

对堆顶进行一次调整算法

第四步2与1交换，堆尾隐式出堆

此时已经有序

代码

public class HeapSort {
    public static void main(String[] args) {
        int [] arr = {7,1,2,3,5,3,6,1};
        heapSort(arr);
        for (int i :arr) {
            System.out.print(i+" ");
        }
    }

    /**
     *
     * @param arr
     * @param k 当前筛的节点的小标
     * @param m 最后一个元素的小标
     */
    private static void sift(int[] arr, int k, int m){
        int i = k;//i 指向被筛选节点
        int j=2*i+1;//j 指向左右孩子较大的节点
        while(j<=m){
            if(j<m&&arr[j]<arr[j+1]){//说明右孩子较大
                ++j;
            }
            if(arr[j]<arr[i])break;
            swap(arr,i,j);
            i=j;
            j=2*i+1;
        }
    }
    public static  void swap(int[] arr, int i,int j){
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }
    private static void heapSort(int[] arr) {
        //将数组调整为最大堆
        for(int i=arr.length/2-1;i>=0;i--){
            sift(arr,i,arr.length-1);
        }
        for(int i=0;i<arr.length;i++){
            //堆顶和堆尾交换
            swap(arr,0,arr.length-i-1);
            //对堆顶进行调整，这里包含了隐含的出堆
            sift(arr,0,arr.length-i-2);
        }
    }
}