堆排序(Heap Sort)
指利用堆这种数据结构所设计的一种排序算法。堆积是一个近似完全二叉树的结构,并同时满足堆积的性质:即子结点的键值或索引总是小于(或者大于)它的父节点。
算法描述
将初始待排序关键字序列(R1,R2….Rn)构建成大顶堆,此堆为初始的无序区;
将堆顶元素R[1]与最后一个元素R[n]交换,此时得到新的无序区(R1,R2,……Rn-1)和新的有序区(Rn),且满足R[1,2…n-1]<=R[n];
由于交换后新的堆顶R[1]可能违反堆的性质,因此需要对当前无序区(R1,R2,……Rn-1)调整为新堆,然后再次将R[1]与无序区最后一个元素交换,得到新的无序区(R1,R2….Rn-2)和新的有序区(Rn-1,Rn)。不断重复此过程直到有序区的元素个数为n-1,则整个排序过程完成。
算法分析
最佳情况:T(n) = O(nlogn) 最差情况:T(n) = O(nlogn) 平均情况:T(n) = O(nlogn)
算法实现
public static void heapSort(int[] arr) {
if (arr == null || arr.length < 2) {
return;
}
for (int i = 0; i < arr.length; i++) {
//0-i之间形成大根堆
heapInsert(arr, i);
}
int size = arr.length;
//从最后一个元素开始操作,1——size是堆的范围
swap(arr, 0, --size);
while (size > 0) {
//一次取出最后的一个节点
heapify(arr, 0, size);
swap(arr, 0, --size);
}
}
public static void heapInsert(int[] arr, int index) {
//循环给左右孩子调整为大根堆
//(0-1)/2=0
while (arr[index] > arr[(index - 1) / 2]) {
swap(arr, index, (index - 1) / 2);
index = (index - 1) / 2;
}
}
//1——heapsize范围是堆的范围
public static void heapify(int[] arr, int index, int heapSize) {
int left = index * 2 + 1;
while (left < heapSize) {
//左右孩子节点中哪一个较大
int largest = left + 1 < heapSize && arr[left + 1] > arr[left] ? left + 1 : left;
//较大的和头结点比较
largest = arr[largest] > arr[index] ? largest : index;
//如果最大是当前头,直接结束循环
if (largest == index) {
break;
}
swap(arr, largest, index);
//指针指向最大的头结点
index = largest;
//继续向左孩子寻找是否有节点大于当前节点
left = index * 2 + 1;
}
}
public static void swap(int[] arr, int i, int j) {
int tmp = arr[i];
arr[i] = arr[j];
arr[j] = tmp;
}
动态图参考:
https://blog.csdn.net/hellozhxy/article/details/79911867