acwing 838堆排序

NirvanaCensor

于 2023-01-17 23:04:25 发布

阅读量235

点赞数

CC 4.0 BY-SA版权

分类专栏： acwing刷题笔记文章标签： c++

本文链接：https://blog.youkuaiyun.com/NirvanaCensor/article/details/128721657

acwing刷题笔记专栏收录该内容

21 篇文章

订阅专栏

该程序展示了如何使用大根堆和小根堆进行升序和降序排序。它首先构建一个大根堆，然后每次将最小元素移动到堆顶并重新调整堆，直到所有元素都处理完毕。同时提供了小根堆的上升操作代码注释。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一般情况下升序用大根堆，降序用小根堆

#include <iostream>
using namespace std;
const int N = 1e5 + 10;

int h[N],cnt = 0;
int n ,m;

void down(int u)
{
    int min = u;//假设当前位置为最小值
    if(u * 2 <= cnt && h[u * 2] < h[min])//没超范围下和左孩子比较
    min = u * 2;
    if(u * 2 + 1 <= cnt && h[u * 2 + 1] < h[min])//没超范围下和右孩子比较
    min = u * 2 + 1;
    if(u != min)//存在比原先节点更小的节点
    {
        swap(h[u],h[min]);
        down(t);//移到t所在节点进行下一轮的down操作
    }
}

int main()
{
    cin >> n >> m;
    for(int i = 1 ;i <= n; i++)
    scanf("%d",&h[i]);
    
    cnt = n;//堆大小
    
    for(int i = n/2; i ; i--)//从下往上做，由于c++的除法是向下取整，无论是左孩子(2n)还是右孩子(2n+1)，/2后都是父节点n
    {
        down(i);
    }
    
    while(m--)
    {
        printf("%d ",h[1]);//出最小
        h[1] = h[cnt--];//最后一个数覆盖第一个数
        down(1);//重新排一遍，将最小的顶上去
    }
    printf("\n");
    return 0;
}

/*小根堆用
void up(int u) 
{
    while(u / 2 > 0 && h[u / 2] >= h[u]) //父节点比子节点大，下沉
    {
        swap(h[u / 2], h[u]);
        u /= 2;
    }
}
*/