java 数据结构之堆排序（HeapSort）详解及实例

java 数据结构之堆排序（HeapSort）详解及实例

1 堆排序

堆是一种重要的数据结构，分为大根堆和小根堆，是完全二叉树, 底层如果用数组存储数据的话，假设某个元素为序号为i(java数组从0开始,i为0到n-1),如果它有左子树，那么左子树的位置是2i+1，如果有右子树，右子树的位置是2i+2，如果有父节点，父节点的位置是(n-1)/2取整。最大堆的任意子树根节点不小于任意子结点，最小堆的根节点不大于任意子结点。

所谓堆排序就是利用堆这种数据结构的性质来对数组进行排序，在数组的非降序排序中，需要使用的就是大根堆，因为根据大根堆的性质可知，最大的值一定在堆顶。堆排序一种不稳定的排序算法，其时间复杂度为O(nlogn)。

2 算法思想

（1）构建最大堆；

（2）选择顶，并与第0位置元素交换；

（3）由于步骤（2）的的交换可能破环了最大堆的性质，即第0位置的元素不再是最大元素，则需要调用maxHeap调整堆(沉降法)，根据实际情况重复步骤（2）。

堆排序中最重要的算法就是maxHeap，该函数假设一个元素的两个子节点都满足最大堆的性质(即左、右子树都是最大堆)，只有根元素可能违反最大堆性质，那么skdNfjYBE把该元素以及左右子节点的最大元素找出来，如果该元素已经最大，那么整棵树都是最大堆，程序退出，否则交换根元素与最大元素的位置，继续调用maxHeap构建最大元素所在的子树。

3 Java代码

public class HeapSort {

public static void main(String[] args) {

int[] arr = {3, 2, 1, 0, -1, -2, -3};

System.out.println("Before heap:");

printArray(arr);

heapSort(arr);

System.out.println("After heap sort:");

printArray(arr);

}

public statichttp:// void heapSort(int[] arr) {

if (arr == null || arr.length <= 1) {

return;

}

buildMaxHeap(arr); //构建最大堆

for (int i = arr.length - 1; i >= 1; i--) {

exchangeElements(arr, 0, i); //交换堆顶和第0位置元素

maxHeap(arr, i, 0); //因为交换元素后，有可能违反堆的性质，所以沉降元素

}

}

private static void buildMaxHeap(int[] arr) { //构建最大堆

if (arr == null || arr.length <= 1) {

return;

}

int half = arr.length / 2;

for (int i = half; i >= 0; i--) {

maxHeap(arr, arr.length, i);

}

}

privahttp://te static void maxHeap(int[] arr, int heapSize, int index) {

int left = index * 2 + 1; //左子树上的元素

int right = index * 2 + 2; //右子树上的元素

int largest = index; //初始化最大元素

if (left < heapSize && arr[left] > arr[index]) {

largest = left;

}

if (right < heapSize && arr[right] > arr[largest]) {

largest = right;

}

if (index != largest) { //判断根元素是否为最大元素

exchangeElements(arr, index, largest);

maxHeap(arr, heapSize, largest);

}

}

public static void printArray(int[] arr) { //打印数组

System.out.print("{");

for (int i = 0; i < arr.length; i++) {

System.out.print(arr[i]);

if (i < arr.length - 1) {

System.out.print(", ");

}

}

System.out.println("}");

}

public static void exchangeElements(int[] arr, int index1, int index2) { //交换元素

int temp = arr[index1];

arr[index1] = arr[index2];

arr[index2] = temp;

}

}

感谢阅读，希望能帮助到大家，谢谢大家对本站的支持！

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。