1.思路
(升序排序)
1.将传入的数组处理成大顶堆。
2.将堆顶元素与末尾元素进行交换,使末尾元素最大。
3.得到第一大的数后,再重复1,2操作得到第二大,第三大……循环处理。最终排序完成。
2.代码
1.如何将数组处理成大顶堆(从下至上,从左至右开始处理)
/**将以i对应的非叶子节点的数,调整成大顶堆
* @param arr 待调整的数组
* @param i 表示非叶子结点在数组中的索引
* @param length 表示对多少个元素进行调整,length在逐渐减少
*/
public static void adjustHeap(int[] arr, int i, int length) {
int temp = arr[i];//先取出当前元素的值,保存在临时变量
//说明:2 * i + 1 是i的左子节点
for(int k = 2 * i + 1; k < length; k = 2 * k + 1) {
if(k + 1 < length && arr[k] < arr[k+1] ) {//此时左子节点小于右子节点
//k + 1 < length 必须写在前面判断
k++;//k指向右子节点
}
if(arr[k] > temp) {//如果子节点大于父节点,
arr[i] = arr[k];//把较大值赋给当前节点
i = k;//i指向k,继续循环比较,防止右子节点下面还有节点
} else {
break;
}
}
//for循环结束后,已经局部将这棵树调整成了一个大顶堆(以i为父节点的最大值处理好了)
arr[i] = temp;//将temp值放到调整后的位置
}
2.排序方法
public static void heapSort(int arr[]) {
//1.先调整成大顶堆
for(int i = arr.length / 2 - 1; i >= 0; --i) {
adjustHeap(arr,i,arr.length);
}
//2.将堆顶元素与末尾元素交换,将最大元素沉到数组末端
int temp;
for(int j = arr.length - 1; j > 0; j--) {
//交换
temp = arr[0];
arr[0] = arr[j];
arr[j] = temp;
//3.重新调整结构,使其满足堆定义,然后继续交换,直至成为有序序列
adjustHeap(arr, 0, j);//这里传0,是因为此时只变动了堆顶元素和末尾元素,其他的不用去管,所以从堆顶开始处理。
}
//System.out.println(Arrays.toString(arr));//为测试暂时不输出排好的数组,因为太多了。。。
3.测试时间
public static void main(String[] args) {
//升序排序
int[] arr = {4,6,8,5,9};
//测试时间:800万2,3s左右
int[] arr1 = new int[8000000];
for(int i = 0; i < 8000000; i++) {
arr1[i] = (int) (Math.random() * 8000000);
}
Date data1 = new Date();
SimpleDateFormat sdf = new SimpleDateFormat("yyyy=MM-dd HH:mm:sss");
String s1 = sdf.format(data1);
System.out.println("排序前时间:"+s1);
heapSort(arr1);
Date data2 = new Date();
String s2 = sdf.format(data2);
System.out.println("排序后时间:"+s2);
}
代码来自:b站韩顺平老师数据结构视频。