堆排序

class HeapSort

package DataStructures.sort;

import java.util.Arrays;

/**
 * 堆排序
 * @author coffee
 * @Date 2021-04-04 12:51
 */
public class HeapSort {

    public static void main(String[] args) {
        int[] arr = {4,6,8,5,9};
        heapSort(arr);
    }

    public static void heapSort(int[] arr){
        int temp;
        //变成堆
        for (int i = arr.length / 2 - 1; i >= 0; i--) {
            adjustHeap(arr,i,arr.length);
        }
        //反复将堆顶元素与末尾元素进行交换
        for (int i = arr.length - 1; i > 0; i--) {
            temp = arr[i];
            arr[i] = arr[0];
            arr[0] = temp;
            adjustHeap(arr,0,i);
        }
        System.out.println(Arrays.toString(arr));
    }

    /**
     * 将一个数组（二叉树）变成一个大顶堆
     * @param arr 数组
     * @param i 非叶子节点在数组中的索引
     * @param len 调整多少个元素，逐渐减少
     */
    public static void adjustHeap(int[] arr,int i,int len){
        int temp = arr[i];

        for (int j = i * 2 + 1; j < len; j = j * 2 + 1) {
            if (j + 1 < len && arr[j] < arr[j + 1]){//左子节点小于右子节点
                j++;//j索引到右子节点
            }
            if (arr[j] > temp){//如果右子节点大于当前节点
                arr[i] = arr[j];//将右子节点赋值给当前节点
                i = j;//主要是确定右子节点的位置
            }else {
                break;
            }
            arr[j] = temp;//将当前节点辅助给右子节点
        }
    }
}