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
| public class Solution { public static void main(String[] args) { int[] nums = {1, 2, 5, 9, 2, 3, 6, 8, 1}; hsort(nums, nums.length); System.out.println(Arrays.toString(nums)); }
private static void hsort(int[] nums, int len) { for (int i = len / 2 - 1; i >= 0; i--) { heapify(nums, i, len); } for (int i = len - 1; i > 0; i--) { swap(nums, i, 0); heapify(nums, 0, i); } } private static void heapify(int[] nums, int cur, int len) { int left = cur * 2 + 1; int right = cur * 2 + 2; int largest = cur; if (left < len && nums[left] > nums[largest]) { largest = left; } if (right < len && nums[right] > nums[largest]) { largest = right; } if (largest != cur) { swap(nums, cur, largest); heapify(nums, largest, len); } }
private static void swap(int[] nums, int i, int j) { int temp = nums[i]; nums[i] = nums[j]; nums[j] = temp; }
}
|