目的:通过建立堆的思想,来让一组序列变成有序数列。
具体操作过程如下:
1.将原序列建立成堆,(升序建立大根堆,降序建立成小根堆)
2.通过不断的取出堆中根结点的元素,将其放入到最后一个位置。再将剩下的元素重新进行堆的排序(即向下调整)。
heapSort调用createHeap函数,来使得传入的数组array变成一个堆。后面的循环,是为了每次都将这个堆的根节点取出,放入数组的最后一个位置,再将剩下元素进行向下转型调整,使得最大的元素又变成根结点。再将这个结点放入数组倒数第二个位置,依次类推。
升序建立大根堆,降序建立成小根堆。就是因为要将根结点位置存入到数组最后,所以如果根结点是最大值,那么放入到数组最后位置以后,该序列就会变成升序。如果根结点是最小值,那么放入数组最后位置以后,该序列就会变成降序。
public void heapSort(int[] array){
createHeap(array);
int end = array.length-1;
while (end > 0){
swap(array,0,end);
shiftDown(array,0,end);
end--;
}
}
将一个数组变成堆的方法,是通过不断的寻找父节点的索引来创建的,通过找到的父节点的索引,从而找到这颗树,再对这颗树进行向下调整的操作。
private void createHeap(int[] array) {
for (int parent = (array.length-1-1)/2;parent>=0;parent--){
shiftDown(array,parent,array.length);
}
}
向下调整的操作,通过父节点,左孩子结点、右孩子结点的值来比大小,交换位置的操作,使得父亲结点都是较大的值。
private void shiftDown(int[] array, int parent, int length) {
int child = 2 * parent + 1;
while (child < length){
if (child + 1 < length && array[child] < array[child+1]){
child++;
}
if (array[child] > array[parent]){
swap(array,child,parent);
parent = child;
child = 2 * parent +1;
}else {
break;
}
}
}