堆排序(Java实现)

堆排序(Java实现)_第1张图片
堆排序(Java实现)_第2张图片

package Tree;

import java.util.Arrays;

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

    /**
     *
     * @param arr 要传入的数组
     */
    public static void heapSort(int[] arr){
        int temp = 0;
        System.out.println("堆排序后的数组:");
        for (int i = arr.length/2-1; i >= 0 ; i--) {
            adjustHeap(arr,i,arr.length);
        }
        for (int j = arr.length -1;j > 0;j --){
            temp = arr[j];
            arr[j] = arr[0];
            arr[0] = temp;
            adjustHeap(arr,0,j);
        }
        System.out.println(Arrays.toString(arr));
    }

    /**
     *
     * @param arr 要进行调整的数组
     * @param i 表示非叶子节点在数组中的索引
     * @param length 表示对多少个元素继续调整
     */
    //将一个二叉树(数组),调整为一个大顶堆
    //思路:首先要保证查找最大元素的顺序为从下到上,比较左右子树的大小,如果找出大的和当前局部父节点进行交换,
    //如果当前左右子树全部小于当前父节点,那么当前父节点即为最大值,因为整棵局部树已经满足堆了,左右子树下面也不会有大于
    //当前父节点的元素
    public static void adjustHeap(int[] arr,int i,int length){
        int temp = 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] > temp){
                arr[i] = arr[k];
                i = k;
            }else{
                break;
            }
        }
        arr[i] = temp;
    }
}

你可能感兴趣的:(DataStructure)