堆排序是一种树形选择排序,是对直接选择排序的有效改进。堆的定义如下:具有n个元素的序列 (h1,h2,...,hn),当且仅当满足(hi>=h2i,hi>=2i+1)或(hi<=h2i,hi<=2i+1)(i=1,2,...,n/2)时称之 为堆。在这里只讨论满足前者条件的堆。由堆的定义可以看出,堆顶元素(即第一个元素)必为最 大项(大顶堆)。完全二叉树可以很直观地表示堆的结构。堆顶为根,其它为左子树、右子树。初 始时把要排序的数的序列看作是一棵顺序存储的二叉树,调整它们的存储序,使之成为一个堆,这 时堆的根节点的数最大。然后将根节点与堆的最后一个节点交换。然后对前面(n-1)个数重新调整使 之成为堆。依此类推,直到只有两个节点的堆,并对它们作交换,最后得到有 n个节点的有序序 列。从算法描述来看,堆排序需要两个过程,一是建立堆,二是堆顶与堆的最后一个元素交换位 置。所以堆排序有两个函数组成。一是建堆的渗透函数,二是反复调用渗透函数实现排序的函数。
建堆 交换 再交换
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;
swap(arr, 0, --size);//将数组末尾的数与堆顶进行交换
while (size > 0) {
heapify(arr, 0, size);//重新调整成大根堆
swap(arr, 0, --size);
}
}
public static void heapInsert(int[] arr, int index) {
while (arr[index] > arr[(index - 1) / 2]) {//若当前位置的值大于父位置的值
swap(arr, index, (index - 1) / 2);//交换
index = (index - 1) / 2;//来到父位置,继续while
}
}
//若数组中的数据进行了变化,则对其进行调整
public static void heapify(int[] arr, int index, int size) {
int left = index * 2 + 1;//左孩子
while (left < size) {
//left+1为右孩子的位置
//取两个孩子中较大值的下标
int largest = left + 1 < size && 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;
}