public class DuiPaiXu {
public static void main(String[] args) {
int[] nums = {3, 7, 4, 9, 7, 2, 1, 6};
heapSort(nums);
System.out.println(Arrays.toString(nums));
}
public static void heapSort(int[] nums){
int n = nums.length;
// 将整个数组构造成大顶堆(从最后一个非叶子结点开始调整)
for(int i = n / 2 - 1; i >= 0; i--){
sink(nums, i, n);
}
while(n > 0){
// 将最大的元素 nums[0] 和 最后一个元素 nums[n - 1] 交换位置
swap(nums, 0, n - 1);
// 将最大的元素删除
n--;
// 修复堆。交换了一次后堆已经不符合条件了,需要重新让它变成大顶堆
sink(nums, 0, n);
}
}
public static void sink(int[] nums, int k, int n){
// 如果沉到堆底,就沉不下去了
while(2 * k + 1 <= n - 1){
// 假设当前结点的左子结点较大。注:bigger 存储的是下标
int bigger = 2 * k + 1;
// 如果当前结点的右子结点存在 && 当前结点的左子结点小于当前结点的右子结点
if(bigger + 1 < n && nums[bigger] < nums[bigger + 1]){
bigger++;
}
// 如果当前结点比它的左右结点中较大的那个还大,那么就不必下沉
// 也就是说如果当前节点比它的左右结点都大,那么就不必下沉
if(nums[k] >= nums[bigger]){
break;
}
// 否则,需要下沉当前节点
swap(nums, k, bigger);
k = bigger;
}
}
public static void swap(int[] nums, int i, int j){
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}
}
堆排序实现
于 2022-01-20 21:45:50 首次发布