java实现堆排序

堆排序

思路：

1.先将数组调整为大顶堆(从小到大排列)或小顶堆(从大到小排列)
(1)遵循从下至上,从左至右的原则,所以从最后一个非子节点开始,对应的索引为:index=arr.length/2-1
(2)需要拿该节点的值,与其子节点进行比较,并将较大值,放入当前位置
(2.1)若交换了,就需要将该子节点下的树,再次调整为大顶堆
(2.2)若没有交换,则可以直接退出循环
2.把顶上的数,与最后的数进行交换,同时将下次进行大顶堆操作的数组的长度减一,因为最大的数已经找到,并放入正确的位置了
3.再次进行大顶堆,这次就可以从上至下了(就是从索引为0开始),因为根节点下的子树都已经满足了大顶堆

public class HeapSort {

    public static void main(String[] args) {
        int[] arr = {1, 5, 3, 4, 9, 6, 0, 12};
        sort(arr);
        System.out.println(Arrays.toString(arr));
    }

    public static void sort(int[] arr) {
        //先将数组置为一个大顶堆
        int count = 0;
        for (int i = arr.length / 2 - 1; i >= 0; i--) {
            adjustHeap(arr, i, arr.length);
            System.out.println("第"+(++count)+"次"+Arrays.toString(arr));
        }

        int temp;
        for (int i = arr.length - 1; i > 0; i--) {
            //交换,将最大的数放到最后
            temp = arr[i];
            arr[i] = arr[0];
            arr[0] = temp;
            //再次进行大顶堆的操作,从顶开始,因为下面的子树都满足大顶堆
            adjustHeap(arr, 0, i);
        }
    }


    /**
     * 操作index节点的子树,将其变为大顶堆
     *
     * @param arr    要操作的数组
     * @param index  从index开始
     * @param length 操作的长度
     */
    public static void adjustHeap(int[] arr, int index, int length) {
        //定义零时变量存放该节点的数
        int temp = arr[index];
        //index*2+1表示的是index节点的左子节点
        for (int k = index * 2 + 1; k < length; k = k * 2 + 1) {
            //若左子节点小于右子节点,则把k+1,让其指向右子节点
            //确保k指向的节点为index节点的子节点中大的一边
            if (k + 1 < length && arr[k] < arr[k + 1]) {
                k++;
            }
            //若子节点中较大的节点大于当前节点,则把较大的节点放到index位置
            //将k赋值给index,使得index每次都是指向temp应该放的位置
            //每次都和temp比较,所以可以不用交换位置,只进行赋值
            if (arr[k] > temp) {
                arr[index] = arr[k];
                index = k; //从该子节点开始,再把下面的树也置为大顶堆
            } else {
                //下面的子树本来已经是大顶堆了,若没有进行交换,则可以直接退出循环
                //因为,该排序是从下至上,从左至右进行,
                //所以操作上面的节点时,下面子树已经是大顶堆
                break;
            }
        }
        //循环结束后,index指向的位置就是temp要放置的位置
        arr[index] = temp;

    }

}