Java堆排序

package com.example.test_java.sort;

import java.util.Arrays;

/**
 * Created by zhu on 2019/3/6.
 * 堆排序
 * <p>
 * 堆是一棵顺序存储的完全二叉树。完全二叉树中所有非终端节点的值均不大于（或不小于）其左、右孩子节点的值。
 * 其中每个节点的值小于等于其左、右孩子的值，这样的堆称为小根堆；
 * 其中每个节点的值大于等于其左、右孩子的值，这样的堆称为大根堆；
 */

public class HeapSort {
    private int A[];
    private int heapSize;

    protected int parent(int i) {
        return (i - 1) / 2;
    }

    protected int left(int i) {
        return 2 * i + 1;
    }

    protected int right(int i) {
        return 2 * i + 2;
    }

    protected void maxHeapify(int i) {
        int l = left(i);
        int r = right(i);
        int largest = i;
        if (l <= heapSize - 1 && A[l] > A[i])
            largest = l;
        if (r <= heapSize - 1 && A[r] > A[largest])
            largest = r;
        if (largest != i) {
            int temp = A[i];
            // swap
            A[i] = A[largest];
            A[largest] = temp;
            this.maxHeapify(largest);
        }
    }

    /**
     * 构建堆的过程：
     * 1、从最后一个非叶子结点开始，然后比较该结点和子节点的大小，结点最大的值来替换该结点的值；
     * 2、如果子节点的值大于该结点的值，则比较子节点和它子节点的大小，结点最大的值来替换该结点的值
     *
     */
    public void buildMaxHeap(int[] A) {
        this.A = A;
        this.heapSize = A.length;

        for (int i = parent(heapSize - 1); i >= 0; i--) {
            maxHeapify(i);
        }
        System.out.println(".............建大根堆...............");

        for (int j = 0; j < A.length; j++) {
            System.out.println("A[" + j + "] = " + A[j]);
        }

    }


    /**
     * 大根堆建完后，最大值在根顶
     * 堆排序步骤：
     * 1.用根顶元素和最后一个叶子结点值替换，那么最后的叶子结点就是最大值了，是有序的了，下次排序不需要动它了
     * 2.从堆顶开始重新创建大根堆，因为替换了A[0]和A[length-1],所以从堆顶开始
     */
    public void heapSort(){
        int step = 1;
        for (int i = A.length - 1; i > 0; i--) {
            int temp = A[i];
            A[i] = A[0];
            A[0] = temp;
            heapSize--;//用根顶元素和最后一个叶子结点值替换，那么最后的叶子结点就是最大值了，是有序的了，下次排序不需要动它了
            System.out.println("Step: " + (step++) + Arrays.toString(A));
            maxHeapify(0);//从堆顶开始重新创建大根堆，因为替换了A[0]和A[length-1],所以从堆顶开始
        }
        System.out.println(".............排序后...............");
        for (int j = 0; j < A.length; j++) {
            System.out.println("A[" + j + "] = " + A[j]);
        }
    }

}

上面都注释了，就不细说了，测试代码如下：

int [] A = {3, 7, 2, 11, 3, 4, 9, 2, 18, 0};
        HeapSort heapSort = new HeapSort();
        heapSort.buildMaxHeap(A);
        heapSort.heapSort();