堆排序算法实现与总结(Java)

堆排序算法

堆排序基本思想

  1. 将待排序序列构造成一个大顶堆
  2. 排序后,整个序列的最大值就是堆顶的根节点。
  3. 将堆顶元素其与末尾元素进行交换,此时末尾元素就为最大值。
  4. 然后将剩余 n - 1 个元素重新构造成一个大顶堆,这样会得到 n 个元素的次小值。如此反复执行,便能得到一个有序序列了

堆排序步骤图解

要求,给定数组{4,6,8,5,9},使用堆排序方法,升序排列


步骤一:构造初始堆。将给定无序序列构造成一个大顶堆(一般升序采用大顶堆,降序采用小顶堆)。

1.假设无序序列原始结构如下:
第一步
2.此时我们从最后一个非叶子结点开始(叶子结点自然不用调整,第一个非叶子结点 arr.length/2-1=5/2-1=1 ,也就是下面的6结点),从左至右,从下至上进行调整。第二步3.找到第二个非叶子节点4,由于[4,9,8]中9元素最大,4和9交换。

第三步4.此时,第三步交换导致了子根[4,5,6]结构混乱,继续调整,[4,5,6]中6最大,交换4和6。
第四步此时,我们就将一个无序序列构造成了一个大顶堆


步骤二:将堆顶元素与末尾元素进行交换,使末尾元素最大。然后继续调整堆,再将堆顶元素与末尾元素交换,得到第二大元素。如此反复进行交换、重建、交换。

1.将堆顶元素9和末尾元素4进行交换
第一步2.重新调整结构,使其继续满足堆定义,此时,堆末尾元素不再参与交换。
第二步3.再将堆顶元素8与末尾元素5进行交换,得到第二大元素8。
第三步4.后续过程重复上面的操作,继续进行调整,交换,如此反复进行,最终使得整个序列有序。
第四步

动态图解
堆排序

代码实现(Java)

package com.athome.sort;

import java.util.Arrays;
import java.util.Random;

/**
 * Description:
 * Author:江洋大盗
 * Date:2021/1/14 12:30
 */
public class HeapSort {
    
    
    /**
     * 此方法将待调整数组调整为符合大顶堆形式的数组
     * @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 = 2*j+1) {
    
    
            //如果左子节点大于右子节点
            if(j+1 < length && arr[j] < arr[j+1]){
    
    
                j++;//指针j指向右子节点
            }
            //如果j指向的值,即i的较大子节点的值大于i指向的父节点的值
            if(arr[j] > temp){
    
    
                arr[i] = arr[j];//就将较大子节点的值赋给父节点
                i = j;//!!!!将i指向j指向的值
            }else{
    
    
                break;
            }
        }
        //当for循环结束后,我们已经将以i为父结点的树的最大值,放在了最顶端(局部)
        arr[i] = temp;
    }

    /**
     * 编写堆排序的方法
     * @param arr 待排序的数组
     */
    public static void heapSort(int[] arr){
    
    
        //1.首先将数组调整为大顶堆的形式
        //注意:终止循环的条件为i >= 0,因为第一个元素即arr[0]肯定是一个非叶子节点,需要调整。
        for (int i = (arr.length-1)/2; i >= 0; i--) {
    
    
            adjustHeap(arr,i,arr.length);
        }
        //2.将堆顶元素与末尾元素交换,将最大元素放置在最后
        //注:循环arr.length-1次即可调整好顺序,最后一个元素位置必然正确,不需要调整。
        int temp = 0;//临时变量,做交换用
        for (int i = arr.length-1; i > 0; i--) {
    
    
            temp = arr[0];
            arr[0] = arr[i];
            arr[i] = temp;
            //3.重新调整结构使其满足堆定义,然后继续交换。
            adjustHeap(arr,0,i);
        }
    }
    //测试程序是否正确
    public static void main(String[] args) {
    
    
        int[] arr = new int[8];
        Random random = new Random();
        for (int i = 0; i < arr.length; i++) {
    
    
            arr[i] = random.nextInt(100);
        }
        System.out.println("排序前:"+ Arrays.toString(arr));
        heapSort(arr);
        System.out.println("排序后:"+ Arrays.toString(arr));

    }
}

测试结果
测试结果


结语

只要能收获甜蜜,荆棘丛中也有蜜蜂忙碌的身影,未来的你一定会感谢现在努力的自己。

猜你喜欢

转载自blog.csdn.net/m0_52099927/article/details/112603676
今日推荐