堆排序是利用堆进行排序的算法。它的基本思想是:
第一步:将待排序列构建成一个最大堆,此时堆顶的根节点就是该序列中的最大值
第二步:将堆顶元素与堆数组的末尾元素交换,此时末尾元素就是最大值,然后将剩余的n-1个序列重新构建成一个堆
如此反复进行第二步,就得到了一个有序序列。下面是代码:
public static void heapSort(int[] arr){
if(arr==null||arr.length==0)
return;
//将现在的待排序列构建成一个最大堆
for(int i=arr.length/2;i>=0;i--){
heapAdjust(arr,i,arr.length);
}
for(int i=arr.length-1;i>0;i--){
swap(arr,0,i);
heapAdjust(arr,0,i);
}
}//取得i的左孩子
private static int getLeftChildIndex(int i){
return 2*i+1;
}
//调整以i为节点的子树为最大堆,n为当前数组最大下标
private static void heapAdjust(int[] arr,int i,int n){
int child;
int tmp;
for(tmp=arr[i];getLeftChildIndex(i)<n;i = child){
child = getLeftChildIndex(i);
if(child<n-1&&arr[child]<arr[child+1]){//找到最大值的孩子节点
child++;
}
if(tmp<arr[child]){
arr[i] =arr[child];
}else
break;
}
arr[i] = tmp;
}
heapAdjust 是核心算法,它的功能是把以下标 i 为初始,长度为 n 的序列构建成最大堆,heapSort中的第一个for
循环的功能是把全部序列构建成一个最大堆,紧接着在第二个for循环中,我们首先被末尾元素与堆顶点元素进行交换
,此时已经把最大元素放置到序列的末尾,即下标 i 处了。接着 执行 heapAdjust(arr,0,i),注意,此时的序列中
除了以下标0为顶点的堆不是最大堆,其他的位置并没有改动,所以只需执行heapAdjust(arr,0,i),就重新构建了
最大堆。反复进行,即排序成功。