堆排序java实现

堆排序是面试中常问的排序方法之一，堆排序的思想是利用堆的性质进行排序
一般来说，升序用大顶堆，降序用小顶堆，下面是小顶堆实现升序排序

public class HeapSort {
    
    /*
    堆排序就是将大顶堆的堆顶元素与最后一个元素交换
    然后将剩余的n - 1个元素建堆
    重复这个过程，最后得到的数组就是排好序的
     */
    public static void sort(int[] arr) {
        for (int i = arr.length / 2 - 1; i >= 0 ; i--) {
            /*
            建堆
             */
            adjustHeap(arr,i,arr.length);
        }
        for (int i = arr.length - 1; i > 0 ; i--) {
            /*
            交换第0个和未排序的最后一个元素
             */
            swap(arr,0,i);
            /*
            维持堆的性质
             */
            adjustHeap(arr,0,i);
        }
    }

    private static void adjustHeap(int[] arr, int i, int length) {
        int tmp = arr[i];
        for (int k = 2 * i + 1; k < length; k = 2 * k + 1) {
            if(k + 1 < length && arr[k] < arr[k + 1]){
                k++;
            }
            if(arr[k] > tmp){
                arr[i] = arr[k];
                i = k;
            }else {
                break;
            }
        }
        arr[i] = tmp;
    }

    public static void swap(int[] arr, int a, int b) {
        int temp = arr[a];
        arr[a] = arr[b];
        arr[b] = temp;
    }
}

测试

	public static void main(String []args){
        int []arr = {7,6,7,11,5,12,3,0,1};
        System.out.println("排序前："+ Arrays.toString(arr));
        sort(arr);
        System.out.println("排序后："+Arrays.toString(arr));
    }

结果

排序前：[7, 6, 7, 11, 5, 12, 3, 0, 1]
排序后：[0, 1, 3, 5, 6, 7, 7, 11, 12]