1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88
| package com.pjmike.sort;
import java.util.Arrays;
public class HeapSort { public static void main(String[] args) { int[] arr = new int[]{2, 3, 2, 5, 8, 4, 9, 0}; create(arr); heapSort(arr); System.out.println(Arrays.toString(arr)); }
static void create(int[] arr) { int n = arr.length; System.out.println("n: " + n); for (int i = n / 2 - 1; i >= 0; i--) { System.out.println("i值:" + i); heapify(arr, n, i); } }
static void heapSort(int[] arr) { int n = arr.length; for (int i = n - 1; i >= 0; i--) { swap(arr, 0, i); heapify(arr, i, 0); } }
static void heapify(int[] arr, int n, int i) { int largest = i; int left = 2 * i + 1; int right = 2 * i + 2; if (left < n && arr[left] > arr[largest]) { largest = left; } if (right < n && arr[right] > arr[largest]) { largest = right; } if (largest != i) { swap(arr, largest, i); heapify(arr, n, largest); } }
public static void swap(int[] arr, int a, int b) { int tmp = arr[a]; arr[a] = arr[b]; arr[b] = tmp; } }
|