堆排序

一--大顶堆和小顶堆

节点的值大于或等于左右孩子节点的值叫大顶堆,节点的值小于或等于其左右孩子的值叫做小顶堆。

 

大顶堆:arr[i] >= arr[2i+1] && arr[i] >= arr[2i+2]  

小顶堆:arr[i] <= arr[2i+1] && arr[i] <= arr[2i+2]  

 

代码 

 

class solution {
    public static void heapSort(int nums) {
        // 先调整
        for (int i = nums.length / 2 - 1; i >= 0; i--){
            adjustHeap(nums, i, length);
        }
        
        // 交换首尾,再调整
        for (int j = nums.length - 1; j >= 0; j--) {
            int temp = nums[j];
            nums[j] = nums[0];
            nums[0] = temp;
            adjustHeap(nums, 0, j);
        }
        
    }
    public static void adjustHeap(int[] nums, int i, int length) {
        int temp = nums[i]; // 保存第一个元素
        // 2 * i + 1 是i的左子节点,k*2+1 是k的左子节点(如果有)
        for (int k = 2 * i + 1; k < length; k = k * 2 + 1) {
            // 找出左右子节点最大的一个
            if (k + 1 < length && nums[k] < nums[k + 1]) {
                k++;
            }
            // 现在开始比较根节点和左右中最大一个谁大
            if (temp < nums[k]) {
                nums[i] = nums[k];
                i = k; // 这里很重要
            } else {
                break; // 因为是从左往右,从下往上调整的,所以敢直接break;
            }    
        }
        nums[i] = temp;
}
}

 

posted on 2021-02-01 13:07  Love&Share  阅读(62)  评论(0编辑  收藏  举报

导航