排序算法之堆排序

堆排序

堆根据其排列特点分为大顶堆和小顶堆。顾名思义，大顶堆是指对于任何一个子树而言，父节

点永远大于它的任何一个子节点。那相反，小顶堆是指对于任何一个子树而言，父节点永远小于它

的任何一个子节点。如下如所示：

我们依次对树中的节点进行编号，映射到数组中就是下面这个样子：

堆排序基本思路：首先，我们将n个元素经过比较且交换位置按关键码建成大顶堆，此时堆顶的元

素（即此时树中的根节点）是最大的，然后我们将堆顶的元素与最后一个节点的元素位置进行交换，

并将此时的最后一个元素放出去，然后在按关键码建成大顶堆，然后再将堆顶的元素与最后一个节点

位置的元素进行交换，依次反复，可以得到一个按关键码有序的序列。这个过程就称之为堆排序。

代码展示

package demo4;

import java.util.Arrays;

public class HeapSort {

    public static void main(String[] args) {
        int[] arr = new int[]{9,6,8,7,0,1,10,4,2};
        heapSort(arr);
        System.out.println(Arrays.toString(arr));
    }

    public static void heapSort(int[] arr){
        //开始位置是最后一个非叶子节点，即最后一个节点的父节点
        int start = (arr.length-1)/2;
        //调整为大顶堆
        for(int i=start;i>=0;i--){
            maxHeap(arr,arr.length,i);
        }
        //先把数组中的第0个和堆中的最后一个数交换位置，再把前面的处理为大顶堆
        for(int i=arr.length-1;i>0;i--){
            int temp = arr[0];
            arr[0] = arr[i];
            arr[i] = temp;
            maxHeap(arr,i,0);
        }
    }

    public static void maxHeap(int[] arr,int size,int index){
        //左子节点
        int leftNode = 2*index+1;
        //右子节点
        int rightNode = 2*index+2;
        int max = index;
        //和两个子节点分别对比，找出最大的节点
        if(leftNode<size&&arr[leftNode]>arr[max]){
            max=leftNode;
        }
        if(rightNode<size&&arr[rightNode]>arr[max]){
            max=rightNode;
        }
        //交换位置
        if(max!=index){
            int temp = arr[index];
            arr[index] = arr[max];
            arr[max] = temp;
            //交换位置以后，可能会破坏之前排好的堆，所以，之前排好的堆要重新调整
            maxHeap(arr,size,max);
        }
    }
}