堆排序的java实现

来源：互联网发布：java 手机短信验证码编辑：程序博客网时间：2024/05/16 01:56

在堆排序的java实现中，待排序的数组arr[0]是空出来不使用的，因为java数组的下标是从0开始的，而堆的起始下标为1（保证满足父节点与左右节点的算数关系）。因此arr[1],arr[2]...ar[arr.length - 1]是被排序的元素集合。
在堆排序算法中，每次将堆的最后一个元素与堆顶元素（当前堆最大元素）进行交换，然后将堆大小heapSize减1，再对根节点进行最大堆的维护，依次类推。这里要注意的是数组的大小是不变的，而堆的大小是变化的，不能简单地认为heapSize=arr.length - 1.正因为此，在最大堆的维护算法和建堆算法中要传入堆大小这个参数。

package mec.lanqiao;import java.util.Arrays;public class Main {static int left(int i) {return i * 2;}static int right(int i) {return i * 2 + 1;}static int parent(int i) {return i / 2;}static void swap(int[] arr, int i, int j) {int temp = arr[i];arr[i] = arr[j];arr[j] = temp;}// 维护最大堆性质，假定以left(i),right(i)为根的二叉树都是最大堆static void maxHeapify(int[] arr, int i, int heapSize) {int l = left(i);int r = right(i);int largest = i;if (l <= heapSize && arr[l] > arr[largest])largest = l;if (r <= heapSize && arr[r] > arr[largest])largest = r;if (largest != i) {swap(arr, i, largest);maxHeapify(arr, largest, heapSize);}}// 建堆static void buildMaxHeap(int[] arr) {int heapSize = arr.length - 1;for (int i = (arr.length - 1) / 2; i > 0; i--) {maxHeapify(arr, i, heapSize);}}// 堆排序static void heapSort(int[] arr, int heapSize) {buildMaxHeap(arr);for (int i = arr.length - 1; i > 1; i--) {swap(arr, 1, i);heapSize--;maxHeapify(arr, 1, heapSize);}}public static void main(String[] args) {int[] arr = { 0, 7, 16, 10, 8, 14, 9, 3, 2, 1, 4 };// maxHeapify(arr, 1, 10);// [16, 14, 10, 8, 7, 9, 3, 2, 4, 1]heapSort(arr, arr.length - 1);System.out.println(Arrays.toString(arr));}}

1 0