1.思路
首先我们来看看什么是原地堆排序?
原地堆排序:原地堆排序就是指在排序过程中不申请多余的存储空间,只利用原来存储待排数据的存储空间用堆这种数据结构进行排序的排序。
用示例来看看吧:
现在给定一个无序数组 [1,3 ,5,8,7,6],将它变成一个升序数组,在不创建新空间的情况下,对它进行原地堆排序。
核心思路:
1.先将数组调整为最大堆
2.不断的将当前堆顶和最后一个元素交换
3.交换之后,无序数组的最大值就放在了最终位置,此时进行siftDown操作
4.重复步骤2和3,当无序数组只剩下一个元素时,整个堆排序完成
步骤1补充知识:升序用大堆,降序用小堆,升序排列时,最大值在数组的最后,所以用大根堆,每次交换元素都是将较大元素放在数组的末尾,降序排列同理
图解:
1.先将数组调整为大根堆
2.此时数组为[8,7,6,5,1,3],将堆顶和数组最后一个元素交换
3.继续将数组调整为大根堆,将堆顶和数组最后一个元素交换
4.重复上述步骤
5.只剩一个元素时,堆排序完成
2.代码
//原地堆排序
public static void heapSort(int[] arr){
//先将数组进行堆化,从非叶子节点的最后一个节点开始
for(int i=(arr.length-1-1)/2;i>=0;i--){
siftDown(arr,i,arr.length);
}
//将堆顶的最大值取出和当前数组的最后一个元素交换
for(int i=arr.length-1;i>0;i--){
swap(arr,0,i);
siftDown(arr,0,i);
}
}
private static void siftDown(int[] arr, int k, int size) {
int j=2*k+1;
//首先保证有子树
while(j>size){
//保证有右子树且右子树的值大于左子树的值
while(j+1>size && arr[j+1]>arr[j]){
j=j+1;
}
//现在索引为j的节点为左右子树的最大值
//现在比较索引为k和索引为j的节点大小
if(arr[k]>arr[j]){
break;
}else{
swap(arr,k,j);
k=j;
j=2*k+1;
}
}
}
public static void swap(int[] arr,int i,int j){
int temp=arr[i];
arr[i]=arr[j];
arr[j]=temp;
}