diff --git a/budd-common/src/main/java/io/github/ehlxr/algorithm/binarytree/heap/Heap.java b/budd-common/src/main/java/io/github/ehlxr/algorithm/binarytree/heap/Heap.java index 5145dc6..4de0e05 100644 --- a/budd-common/src/main/java/io/github/ehlxr/algorithm/binarytree/heap/Heap.java +++ b/budd-common/src/main/java/io/github/ehlxr/algorithm/binarytree/heap/Heap.java @@ -31,26 +31,74 @@ import java.util.Arrays; * @since 2022-03-15 07:40. */ public class Heap { + private static void createHeap(int[] a) { + int n = a.length - 1; // 最后一个叶子节点下标 + buildHeap(a, n); + } + + private static void buildHeap(int[] a, int n) { + // (n - 1) / 2 表示最后一个叶子节点父节点,即为最后一个非叶子节点 + for (int i = (n - 1) / 2; i >= 0; --i) { + heapify(a, n, i); + } + } + + /** + * 堆化 + * + * @param a 数组 + * @param n 最后一个元素下标 + * @param i 需要调整的父节点下标 + */ + private static void heapify(int[] a, int n, int i) { + while (true) { + int max = i; + + // 从当前节点和左右叶子节点中找出最大的 + int l = 2 * i + 1; + if (l <= n && a[l] > a[max]) { + max = l; + } + + int r = 2 * i + 2; + if (r <= n && a[r] > a[max]) { + max = r; + } + + // 当前节点最大时退出 + if (max == i) { + break; + } + + swap(a, i, max); + + // 最大节点和当前节点交换后,继续以当前节点为父节点堆化 + i = max; + } + } + + /** + * 堆排序 + */ + private static void sort(int[] a) { + for (int i = a.length - 1; i > 0; i--) { + buildHeap(a, i); + // 堆顶元素和最后一个元素交换,除过最后一个元素外其它元素再次构建大顶堆 + swap(a, 0, i); + } + } public static void main(String[] args) { int[] arr = new int[]{1, 2, 3, 4, 5, 6}; + createHeap(arr); System.out.println(Arrays.toString(arr)); - sweap(arr,2,3); + + sort(arr); System.out.println(Arrays.toString(arr)); + } - public static int[] heapify(int[] arr) { - - for (int i = arr.length - 1; i > 0; i--) { - if (arr[i] > arr[i / 2]) { - - } - } - - return null; - } - - public static void sweap(int[] arr, int i, int j) { + private static void swap(int[] arr, int i, int j) { int tmp = arr[i]; arr[i] = arr[j]; arr[j] = tmp;