import java.util.Arrays;
public class HeapSort {
/***
* 下滤调整堆(大顶堆)
* 以p为根的数不是大顶堆,但是p节点的左子树和右子树均为大顶堆
* 所以只需将p节点向下交换到合适的位置
***/
public static void adjustHeap(int[] arr, int p, int heapEnd){
int parent = p;
int biggerChild = p * 2 + 1;
while(biggerChild <= heapEnd){
//找出parent节点最大的孩子
if(biggerChild + 1 < heapEnd && arr[biggerChild + 1] > arr[biggerChild])
biggerChild++;
//如果parent比孩子大,则满足大顶堆条件,结束调整
if(arr[parent] >= arr[biggerChild]) break;
//否则交换parent与最大的孩子
int tmp = arr[parent];
arr[parent] = arr[biggerChild];
arr[biggerChild] = tmp;
//从交换节点处继续调整
parent = biggerChild;
biggerChild = parent * 2 + 1;
}
}
/***
* 建堆方法:对初始序列建堆的过程,就是一个反复进行下滤调整的过程。
* 1)n 个结点的完全二叉树,则最后一个结点是第n/2个结点的子树。
* 2)筛选从第n/2个结点为根的子树开始,该子树成为堆。
* 3)之后向前依次对各结点为根的子树进行筛选,使之成为堆,直到根结点。
* ***/
public static void buildHeap(int[] arr, int heapEnd){
for(int i = heapEnd / 2; i >= 0; i--){
adjustHeap(arr, i, heapEnd);
}
}
/***
* 数组 S[]划分为左、右两部分:
* 前缀子数组 S[0..k-1]存放待排序的元素,在算法执行的过程中,它们构成一个大顶堆;
* 后缀子序列 S[k..n-1](依序)存放当前已经排序的元素。
*
* 每次从前缀子数组去除最大值S[0],与S[k-1]交换位置,此时后缀子序列变为S[k-1..n-1],然后继续调整S[0..k-2]为大顶堆
* 每经过一次这样的迭代,堆的规模都将减一,而后缀子序列则会增长一个单元。
* 经过n次迭代之后,大顶堆将完全消失,而后缀子序列则将占据整个序列 S[0..n-1],至此排序任务遂告完成。
* ***/
public static void heapSort(int[] arr){
//初始建堆
buildHeap(arr, arr.length-1);
for(int i = arr.length - 1; i >=0; i--){
//交换堆顶与数组尾部
int tmp = arr[i]; arr[i] = arr[0]; arr[0] = tmp;
//调整堆
adjustHeap(arr, 0, i-1);
}
}
public static void main(String[] args) {
int[] arr = {49,38,65,97,76,13,27,49};
heapSort(arr);
System.out.println(Arrays.toString(arr));
}
}
参考:数据结构与算法(Java 描述)邓俊辉 著