在做一份笔试题时,发现对堆排序算法还不甚了解,所以赶紧了解了一下相关思想和概念,并尝试自身实现了一下,在这附上概念思想以及代码的实现。
堆排序:利用堆数据结构而设计的一种排序算法,堆排序是一种选择排序,它的最坏,最好,平均时间复杂度均为O(nlogn),它也是不稳定排序。首先简单了解下堆结构。堆是具有以下性质的完全二叉树:每个结点的值都大于或等于其左右孩子结点的值,称为大顶堆;或者每个结点的值都小于或等于其左右孩子结点的值,称为小顶堆。
该数组从逻辑上讲就是一个堆结构,用简单公式来描述一下堆的定义就是:
大顶堆:arr[i] >= arr[2i+1] && arr[i] >= arr[2i+2]
小顶堆:arr[i] <= arr[2i+1] && arr[i] <= arr[2i+2]
堆排序思想及步骤:
基本思想:
将待排序序列构造成一个大顶堆,此时,整个序列的最大值就是堆顶的根节点。将其与末尾元素进行交换,此时末尾就为最大值。然后将剩余n-1个元素重新构造成一个大顶堆,这样会得到n个元素的次小值。如此反复执行,便能得到一个有序序列了
步骤:
a.将无需序列构建成一个堆,根据升序降序需求选择大顶堆或小顶堆;
b.将堆顶元素与末尾元素交换,将最大元素"沉"到数组末端;
c.重新调整结构,使其满足堆定义,然后继续交换堆顶元素与当前末尾元素,反复执行调整+交换步骤,直到整个序列有序。
代码实现:
public class HeapSort implements SortableInterface {
@Override
public void sort(int[] array) {
for(int length = array.length ; length > 1 ; length --) {
//从最后一个非叶子节点开始,调整堆的顺序,构建大顶堆
for(int i = length/2 - 1 ; i >= 0 ; i --) {
adjustHeap(array, i , length);
}
//将当前堆的根节点与最后一个叶子节点交换
swap(array , 0 , length -1);
}
for(int i = 0 ; i < array.length ; i++)
System.out.print(array[i] + " ");
}
//调整当前节点及其子节点的位置
public void adjustHeap(int[] array , int i , int length) {
//当当前节点具有左右节点时,需要调整该节点与其子节点的位置
if(2 * i + 1 < length && 2 * i + 2 < length) {
int index = findMax(array , i , 2 * i + 1 ,2 * i + 2);
if(i != index) {
swap(array , i , index);
//重新调整该节点及其子节点的位置,如果有的话
adjustHeap(array , index , length);
}
}else if(2 * i + 1 < length) { //当只具有左字节点时,只调整该节点与左字节点的位置
int index = findMax(array , i , 2 * i + 1 );
if(i != index) {
//交换i位置与index位置的节点
swap(array , i , index);
//重新调整该节点及其子节点的位置,如果有的话
adjustHeap(array , index , length);
}
}
}
//查找最大的节点位置
public int findMax(int[] array , int...indexs) {
//定义最大值的索引,默认为indexs的第一个元素
int index = indexs[0];
for(int j = 1 ; j < indexs.length ; j++) {
//如果当前索引的值大于index索引的值的大小,则目前最大值的索引为当前索引
if(array[index] < array[indexs[j]]) {
index = indexs[j];
}
}
return index;
}
//交换数组中的两个元素
public void swap(int[] array , int index1 , int index2) {
int temp = array[index1];
array[index1] = array[index2];
array[index2] = temp;
}
}
本文概念部分转自:
代码部分全由笔者实现。