八大排序算法----堆排序

堆排序的基本步骤:(以从大到小的顺序排序为例)

1.构建大顶堆(每个结点的值都大于或等于其左右孩子结点的值

2.排序:每次堆顶的元素取出来(整个堆中值最大 ),与最后一个节点做交换,使末尾元素最大

3.交换完之后,需要重新维护堆中剩下的其他节点,保证每次的堆顶都是最大值,重复2,3,直到序列完全有序

Code:

cpp 复制代码
//维护堆的性质
//大顶堆:父节点的左右孩子都比父节点小
//小顶堆:父节点的左右孩子都比父节点大
void heapify(vector<int>& nums, int n, int i)
{
    int large = i;//保存父节点
    int left = 2 * i + 1;//左孩子
    int right = 2 * i + 2;//右孩子
    //判断左孩子是否比父节点大? 大的话,就更新父节点的下标
    if (left<n && nums[left]>nums[large])
        large = left;
    //判断右孩子是否比父节点大? 大的话,就更新父节点的下标
    if (right<n && nums[right]>nums[large])
        large = right;
    //到此,已经找到了当前父节点和其左右孩子中最大的节点的下标
    //判断父节点的下标是否发生变化,如果不相等,说明左右孩子中有比父节点大的
    if (large != i)
    {
        //交换节点,维护大顶堆
        swap(nums[large], nums[i]);
        //继续维护剩下的节点
        heapify(nums, n, large);
    }
}
void heapsort(vector<int>& nums, int n)
{
    //建堆:从最后一个有孩子的父节点开始建立
    //这里为什么是i = n / 2 - 1? 因为左孩子的下标可以表示为2*i+1,此时最后一个孩子的下标为n-1
    //推导过来,找到最后一个有孩子的父节点的下标为n / 2 - 1
    for (int i = n / 2 - 1; i >= 0; i--)
    {
        heapify(nums, n, i);
    }
    //排序:将大顶堆的顶与最后一个叶子节点进行交换,也就是每次找到当前堆中最大的元素,放在数组的最后面
    for (int i = n - 1; i > 0; i--)
    {
        //交换
        swap(nums[i], nums[0]);
        //继续维护大顶堆中剩下节点,要始终保持是大顶堆的顺序
        heapify(nums, i, 0);
    }
}
int main()
{
    int n;
    cin >> n;
    vector<int> nums(n);
    for (int i = 0; i < n; i++)
    {
        cin >> nums[i];
    }
    heapsort(nums, n);
    cout << "按升序顺序排序" << endl;
    for (auto& i : nums)
    {
        cout << i << " ";
    }
    return 0;
}

这里如果要按照从小到大的顺序进行堆排序的话,只需要将维护堆的函数中if判断条件做一点小改动即可。

cpp 复制代码
void heapify(vector<int>& nums, int n, int i)
{
    int small = i;//保存父节点
    int left = 2 * i + 1;//左孩子
    int right = 2 * i + 2;//右孩子
    if (left<n && nums[left]<nums[small])
        small = left;
    if (right<n && nums[right]>nums[small])
        small = right;
    //判断父节点的下标是否发生变化,
    if (small != i)
    {
        //交换节点,维护大顶堆
        swap(nums[small], nums[i]);
        //继续维护剩下的节点
        heapify(nums, n, small);
    }
}

堆排序是不稳定的排序算法。

堆排序的时间复杂度:O(nlogn)

相关推荐
PAK向日葵4 小时前
【算法导论】PDD 0817笔试题题解
算法·面试
地平线开发者7 小时前
ReID/OSNet 算法模型量化转换实践
算法·自动驾驶
快乐的划水a7 小时前
组合模式及优化
c++·设计模式·组合模式
地平线开发者7 小时前
开发者说|EmbodiedGen:为具身智能打造可交互3D世界生成引擎
算法·自动驾驶
星星火柴9368 小时前
关于“双指针法“的总结
数据结构·c++·笔记·学习·算法
艾莉丝努力练剑9 小时前
【洛谷刷题】用C语言和C++做一些入门题,练习洛谷IDE模式:分支机构(一)
c语言·开发语言·数据结构·c++·学习·算法
闪电麦坤9510 小时前
数据结构:迭代方法(Iteration)实现树的遍历
数据结构·二叉树·
C++、Java和Python的菜鸟10 小时前
第六章 统计初步
算法·机器学习·概率论
Cx330❀10 小时前
【数据结构初阶】--排序(五):计数排序,排序算法复杂度对比和稳定性分析
c语言·数据结构·经验分享·笔记·算法·排序算法
散11210 小时前
01数据结构-Prim算法
数据结构·算法·图论