排序5. 堆排序 2021-07-17

最新推荐文章于 2022-12-23 14:44:16 发布

sususuqingxiao

最新推荐文章于 2022-12-23 14:44:16 发布

阅读量78

点赞数 1

分类专栏： c++

本文链接：https://blog.csdn.net/sususuqingxiao/article/details/118857827

版权

c++ 专栏收录该内容

16 篇文章 2 订阅

订阅专栏

排序5. 堆排序

#include <bits/stdc++.h>

using namespace std;

//6堆排序 O(nlogn) 没有最好最坏  空间复杂度O(1) 不稳定 对元素较多的有效
//建堆 调整堆
//从最后一个非叶子结点开始   n/2
void HeapAdjust(vector<int>& nums, int nums_beg, int nums_end)
{
    int temp = nums[nums_beg];//把最后一个调到根处
    for (int j = 2*nums_beg; j <= nums_end; j *= 2)
    {
        if (j < nums_end && nums[j] < nums[j + 1])//大顶堆
            j++;
        if (temp >= nums[j])
            break;
        // nums[s] = nums[j];
        swap(nums[nums_beg], nums[j]);
        nums_beg = j;
    }
    // nums[nums_beg] = temp;
}

void HeapSort(vector<int>& nums)
{
    for (int i = (nums.size() - 1)/2; i >= 1; i--)  //注意这个nums.size()-1 / 2
        HeapAdjust(nums, i, nums.size() - 1);       //建堆

    for (int i = nums.size() - 1; i > 1; i--)
    {
        swap(nums[1], nums[i]); //把根放到最后
        HeapAdjust(nums, 1, i - 1);
    }
}


int main()
{
    vector<int> nums{ -1, 6, 5, 3, 3, 4, 9, 1, 10, 15, 16, 11, 33, 77, 21, 99, 111, 7};
    HeapSort(nums);
    for (auto &x : nums)
        cout << x << ", ";
    cout << endl;
    return 0;
}