堆排序【一种选择排序】

代码

public class Main {
    public static void main(String[] args) {
        int[] arr = {9,8,7,6,5,4,3,2,1};
        sort(arr);

        System.out.println(Arrays.toString(arr));
    }

    public static void sort(int[] arr){
        //完全二叉树 以数组方式存放 则第一个非叶子节点下标为 arr.length/2 - 1
        //首先创建大顶堆
        for (int i = arr.length / 2 - 1;i >= 0;i--){
            adjustHeap(arr,i,arr.length);
        }

        //然后进行堆排序
        for (int j = arr.length - 1;j > 0;j--){
            int temp = arr[j];//把最后一个元素保存起来
            arr[j] = arr[0];
            arr[0] = temp;

            adjustHeap(arr,0,j);
        }
    }

    //大顶堆

    /**
     *
     * @param arr
     * @param i
     * @param length 这个是在堆排序的时候用到的 而不是创建堆的时候用到的
     */
    public static void adjustHeap(int[] arr,int i,int length){
        int temp = arr[i];

        for (int j = 2 * i + 1;j < length;j = j * 2 + 1){
            //先判断左右子节点的大小
            if (j + 1 < length && arr[j] < arr[j + 1]){
                j++;//j指向大的值
            }
            //再判断j指向的子节点和i指向的父节点的大小
            if (temp < arr[j]){//注意 这里是temp与arr[j] 进行比较 而不是arr[i] 和 arr[j]
                //需要给父节点赋值
                arr[i] = arr[j];

                i = j;//将i指向刚更换完的节点
            }else {

                break;
            }
        }

        //将i指向的最后节点赋值
        arr[i] = temp;
    }
}

结果

[1, 2, 3, 4, 5, 6, 7, 8, 9]

完整版【包括小堆排序】

public class Main {
    public static void main(String[] args) {
        int[] arr = {9,8,7,6,5,4,3,2,1};
        sort(arr);
        int[] arr2 = {1,4,3,5,2,8,7,9,6};
        sortSmall(arr2);
        System.out.println(Arrays.toString(arr2));
    }

    //小堆排序
    public static void sortSmall(int[] arr){
        //构建小顶堆
        for (int j = arr.length / 2 - 1;j >= 0;j--){
            adjustSmallHeap(arr,j,arr.length);
        }
        //进行排序
        for (int j = arr.length - 1;j > 0;j--){
            int temp = arr[0];
            arr[0] = arr[j];
            arr[j] = temp;
            //System.out.println("hahah1"+Arrays.toString(arr));
            adjustSmallHeap(arr,0,j);
            //System.out.println("hahah2"+Arrays.toString(arr));
        }
    }
    //构建小顶堆
    public static void adjustSmallHeap(int[] arr,int i,int length){
        int temp = arr[i];

        for (int k = i*2 + 1;k < length;k = k*2 + 1){
            //先让k定位到小的
            if (k + 1 < length && arr[k + 1] < arr[k]){
                k++;
            }
            //然后考虑赋值
            if (temp > arr[k]){
                arr[i] = arr[k];

                i = k;
            }else {
                break;
            }
        }

        arr[i] = temp;//!!!! 注意 这个千万不能忘
    }
    //大堆排序
    public static void sort(int[] arr){
        //完全二叉树 以数组方式存放 则第一个非叶子节点下标为 arr.length/2 - 1
        //首先创建大顶堆
        for (int i = arr.length / 2 - 1;i >= 0;i--){
            adjustHeap(arr,i,arr.length);
        }

        //然后进行堆排序
        for (int j = arr.length - 1;j > 0;j--){
            int temp = arr[j];//把最后一个元素保存起来
            arr[j] = arr[0];
            arr[0] = temp;

            adjustHeap(arr,0,j);
        }
    }

    //大顶堆

    /**
     *
     * @param arr
     * @param i
     * @param length 这个是在堆排序的时候用到的 而不是创建堆的时候用到的
     */
    public static void adjustHeap(int[] arr,int i,int length){
        int temp = arr[i];

        for (int j = 2 * i + 1;j < length;j = j * 2 + 1){
            //先判断左右子节点的大小
            if (j + 1 < length && arr[j] < arr[j + 1]){
                j++;//j指向大的值
            }
            //再判断j指向的子节点和i指向的父节点的大小
            if (temp < arr[j]){//注意 这里是temp与arr[j] 进行比较 而不是arr[i] 和 arr[j]
                //需要给父节点赋值
                arr[i] = arr[j];

                i = j;//将i指向刚更换完的节点
            }else {

                break;
            }
        }

        //将i指向的最后节点赋值
        arr[i] = temp;
    }
}

结果

[9, 8, 7, 6, 5, 4, 3, 2, 1]

一定要注意arr[i] = temp;