ehlxr/budd
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

upgrade heap sort

Browse Source
This commit is contained in:
ehlxr
2020-10-20 10:51:14 +08:00
parent a8aad0afb8
commit 62a0dc3ce1
1 changed files with 19 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

75
src/main/java/me/ehlxr/sort/HeapSort.java
View File

@@ -34,48 +34,21 @@ public class HeapSort {
System.out.println("results: " + Arrays.toString(arr)); System.out.println("results: " + Arrays.toString(arr));
} }
// public static void adjustHeap(int[] arr, int len) {
// if (arr == null || arr.length <= 1 || len <= 1) {
// return;
// }
//
// // 第一个非叶子结点：arr.length/2-1
//
// for (int i = len / 2 - 1; i >= 0; i--) {
// // 找出左、右节点的最大值
// int k = 2 * i + 1;
// if (2 * i + 2 < len && arr[2 * i + 1] < arr[2 * i + 2]) {
// k = 2 * i + 2;
// }
//
// if (arr[k] > arr[i]) {
// swap(arr, i, k);
// }
// }
// }
//
// public static void sort(int[] arr) {
// for (int i = arr.length; i > 1; i--) {
// adjustHeap(arr, i);
//
// System.out.println(Arrays.toString(arr));
// swap(arr, 0, i - 1);
// }
// }
public static void sort(int[] arr) { public static void sort(int[] arr) {
// 1. 构建大顶堆 // 1. 将无序序列构建为一个堆
for (int i = arr.length / 2 - 1; i >= 0; i--) { for (int i = arr.length / 2 - 1; i >= 0; i--) {
// 从第一个非叶子结点从下至上，从右至左调整结构 // 从第一个非叶子结点从下至上，从右至左调整结构
adjustHeap(arr, i, arr.length); adjustHeap(arr, i, arr.length);
} }
// 2. 调整堆结构 + 交换堆顶元素与末尾元素 System.out.println("max heap: " + Arrays.toString(arr));
// 2. 将堆顶元素与末尾元素交换，将最大元素"沉"到数组末端
// 3. 重新调整结构，使其满足堆定义，然后继续交换堆顶元素与当前末尾元素，反复执行调整 + 交换，直到整个序列有序
for (int j = arr.length - 1; j > 0; j--) { for (int j = arr.length - 1; j > 0; j--) {
// 将堆顶元素与末尾元素进行交换
swap(arr, 0, j); swap(arr, 0, j);
// 重新对堆进行调整
adjustHeap(arr, 0, j); adjustHeap(arr, 0, j);
System.out.println(Arrays.toString(arr));
} }
} }
@@ -87,6 +60,8 @@ public class HeapSort {
* @param length 对多少个元素进行调整，length 在逐渐减少 * @param length 对多少个元素进行调整，length 在逐渐减少
*/ */
public static void adjustHeap(int[] arr, int i, int length) { public static void adjustHeap(int[] arr, int i, int length) {
// 先取出当前元素 i
int temp = arr[i];
// 从 i 结点的左子结点开始，也就是 2i+1 处开始 // 从 i 结点的左子结点开始，也就是 2i+1 处开始
for (int k = i * 2 + 1; k < length; k = k * 2 + 1) { for (int k = i * 2 + 1; k < length; k = k * 2 + 1) {
// 如果左子结点小于右子结点，k 指向右子结点 // 如果左子结点小于右子结点，k 指向右子结点
@@ -94,13 +69,20 @@ public class HeapSort {
k++; k++;
} }
// 如果子节点大于父节点，将子节点和父节点交换 // 如果子节点大于父节点，将子节点和父节点交换
if (arr[k] > arr[i]) { // if (arr[k] > arr[i]) {
swap(arr, k, i); // swap(arr, k, i);
if (arr[k] > temp) {
// 不用进行交换
arr[i] = arr[k];
i = k; i = k;
} else { } else {
break; break;
} }
} }
// 将 temp 值放到最终的位置
arr[i] = temp;
} }
public static void swap(int[] arr, int i, int j) { public static void swap(int[] arr, int i, int j) {
@@ -108,25 +90,4 @@ public class HeapSort {
arr[i] = arr[j]; arr[i] = arr[j];
arr[j] = tmp; arr[j] = tmp;
} }
// public static void adjustHeap(int[] arr, int i, int length) {
// // 先取出当前元素 i
// int temp = arr[i];
// // 从 i 结点的左子结点开始，也就是 2i+1 处开始
// for (int k = i * 2 + 1; k < length; k = k * 2 + 1) {
// // 如果左子结点小于右子结点，k 指向右子结点
// if (k + 1 < length && arr[k] < arr[k + 1]) {
// k++;
// }
// // 如果子节点大于父节点，将子节点值赋给父节点（不用进行交换）
// if (arr[k] > temp) {
// arr[i] = arr[k];
// i = k;
// } else {
// break;
// }
// }
// // 将 temp 值放到最终的位置
// arr[i] = temp;
// }
} }