原理:
将数组R[1.....n]看做一颗完全二叉树的顺序存储结构,利用完全二叉树中父节点和子节点之间的内在关系来选择最小的元素。初始将其调整为一个大顶堆,将堆的最后一个元素与堆顶元素交换,则堆的最后一个元素为最大记录(升序), 接着将前n-1个元素重新调整为一个最大堆,再将堆顶元素与当前堆的最后一个元素交换得到次大的记录,重复直到堆中只剩一个元素为止,这个元素就是最小记录。
主要是两个过程: 构建堆 交换堆顶与最后一个元素的位置
大顶堆与小顶堆:
对于给定的n个记录的序列: (r(1) , r(2) , ... , r(n)):
大顶堆: r(i) >= r(2i) && r(i) >= r(2i+1) 此时对顶元素为最大值
小顶堆: r(i) <= r(2i) && r(i) <= r(2i+1) 此时对顶元素为最小值
实现思路:(降序,小顶堆)
pos表示当前的节点所在的索引,则pos的左孩子的索引 child 是 2*pos+1, 找pos,child,child+1中最小的放在pos的位置
初始从二叉树的最后一个非叶子节点开始向上(i = len/2-1)调整最小堆, 然后将堆的第一个元素与当前堆的最后一个元素进行交换,这样每次当前堆的最小的元素始终在顶上。并且每次交换之后都要重新对其余的节点进行调整。
代码:
调整最小堆:
private void adjustMinHeap(int[] a, int pos, int length) {
int temp,child=0; //表示开始的位置(顶),child表示父节点的孩子节点的位置
for(temp=a[pos]; 2*pos+1<=length; pos=child ){
child = 2*pos + 1;
// 另child指向两个孩子中值最小的位置
if(child<length && a[child]>a[child+1]){
child++;
}
if(a[child] < temp){
a[pos] = a[child];
}else
break;
}
a[pos] = temp;
}
交换:
private void myMinHeapSort(int[] a) {
// 首先初始化为最小堆,从最后一个非叶子节点开始向上调整
for(int i=a.length/2-1; i>=0; i--){
adjustMinHeap(a,i,a.length-1);
}
// 将第一个(最小的)与最后一个交换,每次交换之后重新调整(从上到下0,i-1)
for(int j=a.length-1; j>=0; j--){
int tmp = a[0];
a[0] = a[j];
a[j]= tmp;
adjustMinHeap(a, 0, j-1);
}
}
测试代码:
@Test
public void main(){
// int[] a = {4,6,8,5,9};
int[] a = {5,4,9,8,7,6,0,1,3,2};
// 降序
myMinHeapSort(a);
for(int i : a){
System.out.print(i+" ");
}
}