堆排序利用完全二叉树排序,如何将数组转换成完全二叉树,给定的某个节点的下标i,其父节点下标parent[i]、左子节点下标left[i]、右子节点下标right[i]
- parent[i] = (i-1)/2
- left[i] = 2i+1
- right[i] = 2i+2
堆排序有有大根堆和小根堆之分,
- 大根堆:arr[i] >= arr[2i+1] && arr[i] >= arr[2i+2]
- 小根堆:arr[i] <= arr[2i+1] && arr[i] <= arr[2i+2]
大根堆基本思想:将待排序的数组构造成一个大根堆,此时整个数组的最大值就是堆顶的根节点。将其与末尾元素进行交换,此时末尾元素就是最大值。然后将剩下的n-1个元素重新构造成一个大根堆,重复上述步骤,就能得到一个有序数组。
package SortRank;
/**
* 堆排序:基本思想:将无序序列构建成大根堆(此时堆顶元素为序列最大值),再将
* 堆顶元素与末尾元素交换,再将除末尾元素之外的其他元素调整为大根堆,重复以上
* 步骤,直至根节点,就能得到一个有序序列。
* @author 18322
*
*/
public class HeapSort {
public static void main(String[] args) {
HeapSort hs = new HeapSort();
int arr[] = {5, 9, 4, 10, 2, 8, 7};
System.out.println("数组排序前的顺序:");
hs.printArray(arr);
hs.sort(arr);
System.out.println("数组最终排序的顺序:");
hs.printArray(arr);
}
private int[] buildMaxHeap(int[] arr) {
for(int i = arr.length/2 -1; i >= 0; i--) {
adjustHeap(arr, i, arr.length);
}
return arr;
}
private void adjustHeap(int[] arr, int i, int length) {
int temp = arr[i];
for(int k = 2*i + 1; k < length; k = 2*k + 1) {
if(k+1 < length && arr[k] < arr[k+1]) {
k++;
}
if(temp >= arr[k]) {
break;
}else {
arr[i] = arr[k];
i = k;
}
}
arr[i] = temp;
}
private void sort(int[] arr) {
int count = 1;
arr = buildMaxHeap(arr);
for(int i = arr.length-1; i > 0; i--) {
swap(arr, 0 , i);
adjustHeap(arr, 0, i);
System.out.println("第" + count++ + "次排序数组的顺序:");
printArray(arr);
}
}
private void swap(int[] arr, int a, int b) {
int temp = arr[a];
arr[a] = arr[b];
arr[b] = temp;
}
private void printArray(int arr[]) {
for(int in : arr) {
System.out.print(in + "\t");
}
System.out.println();
}
}