堆排序包括两部分,一是数组的堆化,而是交换堆顶元素与最后一个元素的位置。
数组的堆化,需要与归并排序用相同的思想考虑。即都是在原来堆化好的基础上继续堆化。
以下是代码及部分注释:
package sort;
public class heapSort {
// 这部分是在已堆化好的堆上添加pos位置的元素。
public static void adjustMinHeap(int[] a, int pos, int len) {
int temp;
int child;
for (temp = a[pos]; 2 * pos + 1 <= len; pos = child) {//使用temp记录即将插入的元素
child = 2 * pos + 1;
if (child < len && a[child] > a[child + 1]) {//因为插入元素是位于根结点,因此这里选择符合条件的child
child++;
}
if (a[child] < temp) {//选择好子结点之后,做判断,如果成立,将要插入的结点下移。
a[pos] = a[child];
} else
break;
}
a[pos] = temp;//找到符合条件的位置之后,将要已记录的结点放到该位置上。
}
// 这部分有2个功能,一个是堆化,一个是交换元素。
public static void myMinHeapSort(int[] array) {
int i;
int length = array.length;
for (i = length / 2 - 1; i >= 0; i--) {// 该处进行堆化,后一半元素肯定满足堆化条件,从中央部位开始向前对每个元素开始插入堆化
adjustMinHeap(array, i, length - 1);
}
for (i = length - 1; i >= 0; i--) {// 开始交换元素。因为堆化的为一个最小堆,因此得到逆序(即从大到小)。
int tmp = array[0];
array[0] = array[i];
array[i] = tmp;
adjustMinHeap(array, 0, i - 1);
}
}
public static void main(String[] args) {
int i = 0;
int a[] = { 9, 8, 7, 4, 6, 2, 4, 3, 1, 5 };
int len = a.length;
myMinHeapSort(a);
for (i = 0; i < len; i++) {
System.out.print(a[i] + " ");
}
}
}