public class HeapSort3 {
public static void main(String[] args) {
int[] arr = {312,126,272,226,28,165,123,8,12};
// int[] arr = SortTestHelper.getRandomArray(20, 0, 2);
System.out.println("堆排序前:"+Arrays.toString(arr));
//堆排序
heapSort(arr, arr.length);
System.out.println("堆排序后:"+Arrays.toString(arr));
}
private static void heapSort(int[] arr, int count) {
//完成构建大顶堆
//(count-2)/2获取最后一个非叶子节点
for(int i=(count-2)/2; i>=0; i--)
shiftDown(arr, count, i);
System.out.println("构建完成的大顶堆:"+Arrays.toString(arr));
//将大顶堆头结点(最大数)与最后一个元素交换位置,然后除去最后这个最大的元素,再shiftDown操作维护该(除去最后一个元素的数组)堆为大顶堆。循环直到升序排序完成
for(int j=count-1; j>0; j--) {
swap(arr, 0, j);
shiftDown(arr, j, 0);
}
}
/**
*
* @param arr 数组
* @param count 元素个数
* @param currentRoot 当前根节点的下标
*/
private static void shiftDown(int[] arr, int count ,int currentRoot) {
while(2*currentRoot+1 < count) {
int max = 2*currentRoot+1; //初始赋值left孩子
if(max+1 < count && arr[max+1] > arr[max])
max += 1;
if(arr[currentRoot] >= arr[max])
break;
swap(arr, currentRoot, max);
currentRoot = max;
}
}
//交换
public static void swap(int[] arr, int n, int m) {
int temp = arr[n];
arr[n] = arr[m];
arr[m] = temp;
}
}
---------------------
作者:stoneWang_L
来源:CSDN
原文:https://blog.csdn.net/stoneWang_L/article/details/89190250
版权声明:本文为博主原创文章,转载请附上博文链接!
Java排序算法--堆排序
最新推荐文章于 2024-09-03 20:17:53 发布