

堆排序的基本思想




此时,我们就得到了一个大顶堆





package com.atguigu.tree;
import java.util.Arrays;
public class HeapSort {
public static void main(String[] args) {
int[] arr={4,6,8,5,9};
heapSort(arr);
//插入排序的速度测试
int[] arr2=new int[80000];
for (int i=0;i<80000;i++){
arr2[i]=(int)(Math.random()*40000);//生成一个[0,20000)的随机整数
}
long startTime=System.currentTimeMillis();
heapSort(arr2);
long endTime=System.currentTimeMillis();
System.out.println("排序用的时间:"+(endTime-startTime));
}
//编写一个堆排序的方法
public static void heapSort(int[] arr){
int temp=0;
System.out.println("堆排序");
//分部完成
adjustHeap(arr,1,arr.length);
System.out.println("第一次调整后:"+ Arrays.toString(arr));
adjustHeap(arr,0,arr.length);
System.out.println("第二次调整后:"+Arrays.toString(arr));
//完成我们的最终代码
//将一个无序列构建成一个堆,根据升序降序需求选择大顶堆或小顶堆
for(int i=arr.length/2-1;i>=0;i--){
adjustHeap(arr,i,arr.length);
}
System.out.println("数组:"+Arrays.toString(arr));
//将堆顶元素与末尾元素交换,将最大元素"沉"到数组末端
//重写调整结构,使其能满足堆定义,然后继续交换堆顶元素与当前末尾元素,反复调整+交换步骤,直到整个序列有序
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 表示非叶子节点的在数组中索引
* @param length 多少个元素进行调整,length是在逐渐减少
*/
public static void adjustHeap(int[] arr,int i,int length){
int temp=arr[i];//先取出当前元素的值,保存在临时变量里面
//开始调整
/*
说明,k=i*2+1是i节点的左子节点.
*/
for(int k=i*2+1;k<length;k=k*2+1){
if((k+1)<length&&arr[k]<arr[k+1]){//说明左子节点它的值小于右子节点的值
k++;//k指向右子节点
}
if(arr[k]>temp){//如果子节点大于父节点
arr[i]=arr[k];//把较大的值赋值给当前的父节点
i=k;//让i执行k,继续循环比较
}else {
break;
}
}
//当for循环结束后,我们已经将以i为父节点的树的最大值,调整到了最上面
arr[i]=temp;//将temp放到调整之后的位置
}
}