堆排序最好、最坏、平均情况的复杂度都是O(nlgn),是不稳定的,对于在较大的N个数中,找出最大(或者最小)的m个数非常有效。
堆排序算法Java版:
import java.util.Arrays;
public class HeapSort {
private static int N = 10;
public static void main(String[] args) {
int[] arr = new int[N];
for (int i = 0; i < N; i++) {
arr[i] = ((Double) (Math.random() * 100)).intValue(); //使用随机数填充数组
}
System.out.println("排序前的数组 : " + Arrays.toString(arr));
heapSort(arr, N);
System.out.println("排序后的数组 : " + Arrays.toString(arr));
}
public static void heapSort(int[] arr, int n) {
for(int i=n/2-1;i>=0;i--) { //从下往上初始化堆,是满足定义
initHeap(arr, i, n);
}
for(int i=n-1;i>=1;i--) { //依次将堆顶的最大元素与后面的交换,完成排序
swap(arr, 0, i);
initHeap(arr, 0, i); //将次大元素再放置堆顶
}
}
/**
* 初始化调整堆,是满足堆结构定义:父节点不小于子节点
* @param arr 待排数组(堆结构)
* @param start 当前节点
* @param end 最大节点
*/
public static void initHeap(int[] arr, int start, int end) {
int left = 2*start + 1; //当前节点的左孩子
int right = 2*start + 2; //当前节点的右孩子
int index = start; //记录当前、左孩子、右孩子中的最大元素下标
if(left<end && arr[left]>arr[index]) index = left;
if(right<end && arr[right]>arr[index]) index = right;
if(index != start) {
swap(arr, index, start);
initHeap(arr, index, end); //继续向下调整交换后的堆
}
}
/**
* 交换数组arr[i]与arr[j]
*/
public static void swap(int[] arr, int i, int j) {
int temp = arr[i];
arr[i] = arr[j];
arr[j] = temp;
}
}