堆排序

123 阅读1分钟
import java.util.Arrays;

/**
 * 堆排序
 * 个人理解:1.大顶堆:每个子树都是父节点大于子节点
 *  2.先将数组变成大顶堆,然后调整首位元素(值大的放尾部),然后再将除最后元素之外的堆变成大顶堆,以此类推
 *  3.时间复杂度,所以最好情况和最坏情况都是O(n*logn)
 *  3.空间复杂度和插入排序一样,因为没有用的额外空间,所以为0(1)
 *  4.稳定性,因为发生跳跃式变换(跳跃式指非相邻元素交换位置),所以不稳定
 */
public class SortTest {

    public static void main(String[] args) {
        int[] arr = {70, 12, 82, 43, 99, 23, 38, 53, 22};

        heapSort(arr);
        Arrays.stream(arr).forEach(a->System.out.print(a+" "));
    }

    public static void heapSort(int[] arr) {
        //创建大顶堆
        createHeap(arr);

        int end = arr.length - 1;
        while (end > 0) {
            //交换首位元素
            int temp = arr[end];
            arr[end] = arr[0];
            arr[0] = temp;
            //调整堆为大顶堆
            adjustHeap(arr, 0, end);
            end--;
        }
    }

    private static void adjustHeap(int[] arr, int parent, int end) {
        int child = 2 * parent + 1;

        while (child < end) {
            if (child+1<end && arr[child] < arr[child+1]) {
                child++;
            }
            if (arr[child] > arr[parent]) {
                int temp = arr[child];
                arr[child] = arr[parent];
                arr[parent] = temp;
                parent = child;
                child = 2 * parent + 1;
            } else {
                break;
            }
        }

    }

    private static void createHeap(int[] arr) {
        for (int i=(arr.length-1-1)/2; i>=0; i--) {
            adjustHeap(arr, i, arr.length);
        }
    }

}