堆排序是利用堆的性质进行的一种选择排序。下面先讨论一下堆。
1.堆
堆实际上是一棵完全二叉树,其任何一非叶节点满足性质:
Key[i]<=key[2i+1]&& Key[i]<=key[2i+2]
或者
Key[i]>=Key[2i+1]&& key>=key[2i+2]
即任何一非叶节点的关键字不大于或者不小于其左右孩子节点的关键字。堆分为大顶堆和小顶堆,满足Key[i]>=Key[2i+1] && key>=key[2i+2]称为大顶堆,满足
Key[i]<=key[2i+1]&&Key[i]<=key[2i+2]称为小顶堆。由上述性质可知大顶堆的堆顶的关键字肯定是所有关键字中最大的,小顶堆的堆顶的关键字是所有关键字中最小的。
2.堆排序的思想
利用大顶堆(小顶堆)堆顶记录的是最大关键字(最小关键字)这一特性,使得每次从无序中选择最大记录(最小记录)变得简单。
其基本思想为(大顶堆):
1)将初始待排序关键字序列(R1,R2….Rn)构建成大顶堆,此堆为初始的无序区;
2)将堆顶元素R[1]与最后一个元素R[n]交换,此时得到新的无序区(R1,R2,……Rn-1)和新的有序区(Rn),且满足R[1,2...n-1]<=R[n];
3)由于交换后新的堆顶R[1]可能违反堆的性质,因此需要对当前无序区(R1,R2,……Rn-1)调整为新堆,然后再次将R[1]与无序区最后一个元素交换,得到新的无序区(R1,R2….Rn-2)和新的有序区(Rn-1,Rn)。不断重复此过程直到有序区的元素个数为n-1,则整个排序过程完成。
操作过程如下:
1)初始化堆:将R[1..n]构造为堆;
2)将当前无序区的堆顶元素R[1]同该区间的最后一个记录交换,然后将新的无序区调整为新的堆。因此对于堆排序,最重要的两个操作就是构造初始堆和调整堆,其实构造初始堆事实上也是调整堆的过程,只不过构造初始堆是对所有的非叶节点都进行调整。从上述过程可知,堆排序其实也是一种选择排序,是一种树形选择排序。只不过直接选择排序中,为了从R[1...n]中选择最大记录,需比较n-1次,然后从R[1...n-2]中选择最大记录需比较n-2次。事实上这n-2次比较中有很多已经在前面的n-1次比较中已经做过,而树形选择排序恰好利用树形的特点保存了部分前面的比较结果,因此可以减少比较次数。对于n个关键字序列,最坏情况下每个节点需比较log2(n)次,因此其最坏情况下时间复杂度为nlogn。堆排序为不稳定排序,不适合记录较少的排序。
具体实现如下:
/**
* @Title: HeapSort.java
* @Package sortandsearch
* @Description: TODO
* @author peidong
* @date 2017-5-10 上午8:35:44
* @version V1.0
*/
packagesortandsearch;
/**
* @ClassName: HeapSort
* @Description: 堆排序实例
* @date 2017-5-10 上午8:35:44
*
*//*
publicclass HeapSort {
/**
*
* @Title: print
* @Description: 打印数组
* @param arr
* @return void
* @throws
*/
public static void print(int[] arr) {
for (int i = 0; i < arr.length;i++) {
System.out.print(arr[i] +" ");
}
}
/**
*
* @Title: swap
* @Description: 交换函数
* @param i
* @param j
* @return void
* @throws
*/
public static void swap(int[] arr, int i, int j) {
int temp = arr[i]; //此处直接用i,j交换会报错
arr[i] = arr[j];
arr[j] = temp;
}
/**
*
* @Title: heapSort
* @Description: 推排序
* @param arr
* @return void
* @throws
*/
public static void heapSort(int[] arr) {
//边界条件判断
if (arr == null || arr.length<= 1) {
return;
}
//构建最大堆
buildMaxHeap(arr);
for (int i = arr.length - 1; i>= 1; i--) {
swap(arr, 0, i); //交换堆顶和最后一个元素,即每次将剩余元素中的最大者放到最后面
heapAdjust(arr, i, 0); //重新调整堆顶节点成为大顶堆
}
}
/**
*
* @Title: buildMaxHeap
* @Description: 构建最大堆
* @param arr
* @return void
* @throws
*/
private static void buildMaxHeap(int[]arr) {
//边界条件判断
if (arr == null || arr.length<= 1) {
return;
}
//非叶节点最大序号值为size/2
int half = arr.length / 2;
for (int i = half; i >= 0; i--){
heapAdjust(arr, arr.length,i);
}
}
/**
*
* @Title: maxHeap
* @Description: 调整堆
* @param arr
* @param heapSize
* @param index
* @return void
* @throws
*/
private static void heapAdjust(int[] arr,int heapSize, int index) {
//左右节点序号
int left = index * 2 + 1;
int right = index * 2 + 2;
int largest = index;
if (left < heapSize &&arr[left] > arr[index]) {
largest = left;
}
if (right < heapSize &&arr[right] > arr[largest]) {
largest = right;
}
if (index != largest) {
swap(arr, index, largest);
heapAdjust(arr, heapSize,largest); //避免调整之后以max为父节点的子树不是堆
}
}
public static void main(String[] args) {
int[] arr = {1, 3, 5, 7, 9, 2, 4,6, 8, 0, 12, 11, 14, 16 };
System.out.println("原始数组为");
print(arr);
heapSort(arr);
System.out.println();
System.out.println("推排序之后的数组为:");
print(arr);
}
}