堆化:定义:把一个乱序的数组变成一个大顶堆或者小顶堆
堆化对堆排序有重要作用,后面会阐述
如何让给定的一个数组实现堆化:(这里我们以大顶堆为例)
假设这里有个数组arr[]={1,8,5,6,2,4,11}
对应的二叉树的形式
这里我们先找到最后一个非叶子节点(arr.length/2-1),为什么是arr.length/2-1?首先我们知道已知子节点i则父节点:(i-1)/2,而最后一个非叶子节点的子节点一定是arr.length-1。
这里我们找到最后一个非叶子节点5后,然后把 这一小部分
调整成大顶堆,
然后再让前一个节点8那里也变成大顶堆;
接着就是1这个节点 把他也调成大顶堆
因为这里交换了1与11两个节点,所以需要向下调整(通过递归实现)详细见代码部分
至此我们就是通过从下往上逐步实现堆化,调整过后再向下调整。
当我们堆化完成,得到了大顶堆后,将堆顶元素与末尾元素进行交换,交换过后就不是大顶堆了,所以我们还需要向下调整再次变成大顶堆,但是我们需要注意:交换过后已经是想要的结果,所以不能调整它的位置了,所以需要逐渐缩小调整的范围。
import java.util.Arrays;
/**
* 堆排序demo
*/
//堆化的时候用的是大顶堆,所以是升序,反之小顶堆是降序
public class 堆排序 {
public static void main(String []args){
int []arr = {1,8,5,6,2,4,11};
sort(arr);
System.out.println(Arrays.toString(arr));
}
public static void sort(int []arr){
//1.构建大顶堆
for(int i=arr.length/2-1;i>=0;i--){
//从第一个非叶子结点从下至上,从右至左调整结构
adjustHeap(arr,i,arr.length);
}
//2.调整堆结构+交换堆顶元素与末尾元素
for(int j=arr.length-1;j>0;j--){
swap(arr,0,j);//将堆顶元素与末尾元素进行交换
adjustHeap(arr,0,j);//重新对堆进行调整,交换过后已经是想要的结果,所以不能调整它的位置了,所以范围需要逐渐缩小
}
}
/**
* 调整大顶堆(仅是调整过程,建立在大顶堆已构建的基础上)
* @param arr
* @param i
* @param length
*/
public static void adjustHeap(int []arr,int i,int length){
int temp = arr[i];//先取出当前元素i
int left=i*2+1;
int right = i*2+2;
int k;
//说明该节点是叶子节点,left已经越界了
if(left>=length) {//这里也是递归的出口
return;
}
//有左孩子,但是右孩子越界了不存在
else if(right>=length) {
k = left;
}
//孩子节点中左孩子是最大的
else if(arr[left]>=arr[right]){
k = left;
}
else {
k = right;
}
//父节点大于孩子节点,不用调整。
if(arr[i]>=arr[k]) {
return;
}
else {
temp = arr[i];
arr[i] = arr[k];
arr[k] = temp;
adjustHeap(arr,k,length);//父节点与孩子节点交换值过后需要向下调整,所以用递归。
}
}
/**
* 交换元素
* @param arr
* @param a
* @param b
*/
public static void swap(int []arr,int a ,int b){
int temp=arr[a];
arr[a] = arr[b];
arr[b] = temp;
}
}
最后:堆化所用时间是n*lg(n)
堆化后排序所用时间是n*lg(n)