【数据结构】15-堆排序

一、堆排序

二、思想

三、步骤

四、整体代码

package cn.qqqking.tree;


import java.util.Arrays;


/**
* @author AnQi
* @date 2020/4/24 10 30:24
* @description
*/
public class HeapSort {
    public static void main(String[] args) {
        int arr[] = {4,6,8,5,9};


        System.out.println("调整前：");
       System.out.println(Arrays.toString(arr));
       adjustHeap(arr,1,arr.length);
       System.out.println("调整后：");
       System.out.println(Arrays.toString(arr));
       adjustHeap(arr,0,arr.length);
       System.out.println(Arrays.toString(arr));


        //分布完成  第一次调整
    //    heapSort(arr);




    }




    /**
     *  完成堆排序方法
     *  升序排列（大顶堆）
     */
   public static void heapSort(int arr[]){


       int temp = 0 ;




       //final
       for (int i = arr.length / 2; 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));




   }




    /**
     * 完成 将 以i 对应的非叶子节点的树调整成大顶堆
     * 讲一个数组（二叉树） 调整成一个大顶堆
     * @param arr  数组
     * @param i   非叶子节点在数组中的index
     * @param length 表示对多少个元素进行调整
     */


    public static void adjustHeap(int arr[],int i, int length){




        int temp = arr[i];//先取出当前元素的值 保存在临时变量
        //开始调整
        //说明
        //1、 k= i * 2 + 1  k是i的左子节点
        for (int k= i * 2 + 1 ; k < length;k = k * 2 + 1) {


            if(k+1 < length&& arr[k] < arr[k+1]){
                //左子节点 小于 右子节点
                k++;//指向右子节点
            }
            if(arr[k] > temp){//如果子节点大于父节点
                arr[i] =arr[k]; //把较大的值 赋值给当前节点
                i =  k;  // i指向k 循环比较
            }else{
                break;
            }
        }
        //结束后 说明 一个小子树的父节点为最大 注意不是大顶堆
        //目前i为k交换之前的位置 需要将数据替换
        arr[i] = temp;


    }


}