Java编程内功-数据结构与算法「堆排序」

一个数组中非叶子节点的个数 = arr.length / 2 - 1

代码案例

package com.xie.tree; 
 
public class HeapSort { 
    public static void main(String[] args) { 
        int[] arr = new int[8000000]; 
        for (int i = 0; i < 8000000; i++) { 
            arr[i] = (int) (Math.random() * 800000000); 
        } 
        long start = System.currentTimeMillis(); 
        heapSort(arr); 
        long end = System.currentTimeMillis(); 
        System.out.println("耗时：" + (end - start) + "ms"); 
         
    } 
 
    public static void heapSort(int[] arr) { 
        int temp = 0; 
        System.out.println("堆排序!!"); 
 
        //1.将无序序列构成一个堆，根据升序降序需求选择大顶堆或小顶堆 
        for (int i = arr.length / 2 - 1; i >= 0; i--) { 
            adjustHeap(arr, i, arr.length); 
        } 
        //2.将堆顶元素与数组末尾元素交换，将最大元素"沉"到数组末端 
        //3.重新调整结构，使其满足堆定义，然后继续交换堆顶元素与当前末尾元素，反复执行调整+交换步骤，直到整个序列有序。 
        for (int j = arr.length - 1; j > 0; j--) { 
            //交换 
            temp = arr[j]; 
            arr[j] = arr[0]; 
            arr[0] = temp; 
            adjustHeap(arr, 0, j); 
        } 
    } 
 
     
    public static void adjustHeap(int[] arr, int i, int length) { 
        //先取出当前元素的值，保存在临时变量 
        int temp = arr[i]; 
        //k = 2 * i + 1  是i节点的左子节点 
        for (int k = 2 * i + 1; k < length; k = k * 2 + 1) { 
            //当左子节点值小于右子节点值 
            if (k + 1 < length && arr[k] < arr[k + 1]) { 
                k++;//k指向右子节点 
            } 
 
            //如果子节点值大于父节点值 
            if (arr[k] > temp) { 
                //把较大的值赋给当前节点 
                arr[i] = arr[k]; 
                //!!! i指向k 继续循环比较 
                i = k; 
            } else { 
                break; 
            } 
        } 
 
        //当for循环结束后，我们已经将以 i 为父节点的树的最大值，放在了最顶。 
 
        //将temp值放到调整后的位置 
        arr[i] = temp; 
    } 
} 

 【编辑推荐】