- 思路:
1、首先是初始化堆,建立从小到大的升序堆,只需要保证父节点的值大于相邻的子节点值即可
2、将数组收尾换位置,确保最大的元素在数组的尾部。
3、此时从头部元素开始调整堆,使他满足从小到大的序列,以此执行2-3步,直到执行n-1次(n:数组的长度)
-
代码:
package heap;
import java.text.SimpleDateFormat;
import java.util.Arrays;
import java.util.Date;
public class HeapSortDemo {
//升序
public static void main(String[] args) {
// TODO Auto-generated method stub
// int[] arr = {2,4,9,5,1};
// HeapSortDemo heapSortDemo = new HeapSortDemo();
// heapSortDemo.heapSort(arr);
// System.out.println(Arrays.toString(arr));
HeapSortDemo heapSortDemo = new HeapSortDemo();
int[] arr = new int[8000000];
for(int i = 0; i < arr.length; i++) {
arr[i] = (int)(Math.random()*80000000);
}
Date date1 = new Date();
SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
String datasString1 = simpleDateFormat.format(date1);
System.out.println("排序前:"+datasString1);
heapSortDemo.heapSort(arr);
Date date2 = new Date();
String datasString2 = simpleDateFormat.format(date2);
System.out.println("排序后:"+datasString2);
}
public void heapSort(int[] arr) {
if(arr.length == 0) {
return;
}
int temp;
//1、初始化大顶堆
for(int i = arr.length/2-1; i >= 0; i--) {
adjustHeap(arr, i, arr.length);
}
//2、将收尾元素交换,并重新调整堆
for (int i = arr.length-1 ; i > 0; i--) {
// System.out.println(Arrays.toString(arr));
//将收尾元素交换
temp = arr[0];
arr[0] = arr[i];
arr[i] = temp;
//并重新调整堆
adjustHeap(arr, 0, i);
}
}
//构建以i为父节点的大顶推(局部的大顶堆)
/**
*
* @param arr 待排序的数组
* @param i 父节点在数组中的下标
* @param length 数组长度
*/
public void adjustHeap(int[] arr, int i, int length) {
int temp = arr[i];
//左子树节点:k = i*2+1
for(int k = i*2+1; k < length; k = k*2+1 ) {
if(k+1 < length && arr[k+1] > arr[k]) {
k++;
}
if(arr[k] > temp) {
arr[i] = arr[k];
i = k;
}else {
break;
}
}
//将父节点放在合适的位置
arr[i] = temp;
}
}