堆的基本概念
堆分为大根堆和小根堆。堆可以选出最大或最小的值,大根堆根最大,小根堆根最小。
堆的逻辑结构是树的形式,一般存储结构则采用数组形式。
如以下堆:
数组存储结构:
堆排序
![](https://img-blog.csdnimg.cn/20210324102518462.png)
对上述数组进行堆排序的步骤分析:
-
建立初始堆
-
从第一个非叶子节点开始进行堆调整:
调整6:
调整1:
调整5:
调整4:
到这里,已经完成一次排序,每一次排序会找到一个最大值即根元素8。去除该元素对接下来的元素进行排序。
表现在数组结构中,即,每次把最大值调整到数组最后一个元素交换,并且不再参与接下来的排序。 -
第一轮结束,重复2步骤,寻找第二轮最大值。
第三轮排序:
第四轮:
第五轮:
第六轮:
第七轮:
第八轮:
排序完成!
代码实现
1.数组的存储形式
2.节点序号问题:当前节点为i时,二叉树的左子节点为2i+1,右子节点为2i+2。
3. 从第一个非叶子节点开始排序:第一个非叶子节点的序号为(n-1)/2
因为最后一个元素序号为n时,它的父节点即为第一个非叶子节点,父节点的序号为(n-1)/2
package suanfa;
import java.util.Arrays;
public class Heap {
public static void main(String[] args) {
int[] a = {4,5,1,6,2,7,3,8};
System.out.println("排序前:"+Arrays.toString(a));
heapSort(a);
System.out.println("排序后:"+ Arrays.toString(a));
}
public static void heapSort(int[]a){
int len = a.length;
heapSort(a,len-1);
}
public static void heapSort(int[]a,int end){ //end为待排序的最后一个元素的下标
//一次堆排序找到一个最大值
for(int i = (end-1)/2;i>=0;i--){
if(2*i+1<=end){
int min = Math.min(a[i],a[2*i+1]);
int max = Math.max(a[i],a[2*i+1]);
a[i] = max;
a[2*i+1] = min;
}
if(2*i+2<=end){
int min = Math.min(a[i],a[2*i+2]);
int max = Math.max(a[i],a[2*i+2]);
a[i] = max;
a[2*i+2] = min;
}
}
//交换最大值和数组末尾元素,对前面元素进行排序
if(end>0) { //end==0时,已完成排序
int temp = a[0];
a[0] = a[end];
a[end] = temp;
heapSort(a, end - 1); //进行下一轮排序
}
}
}
堆排序的应用
堆排序每轮排序会找到一个最大值或者最小值,可应用于寻找前k个最大值或最小值,当找到K个就停止,可以降低复杂度
求中位数