堆排序

#include<bits/stdc++.h>
using namespace std;
void swap(int tree[],int i,int j)
{
    int temp = tree[i];
    tree[i] = tree[j];
    tree[j] = temp;
}
void heapify(int tree[],int n,int i)
{
    if(i>=n)
        return ;
    int c1 = 2*i+1;
    int c2 = 2*i+2;
    int maxx = i;
    if(c1<n&&tree[c1]>tree[maxx])
    {
        maxx = c1;
    }
    if(c2<n&&tree[c2]>tree[maxx])
    {
        maxx = c2;
    }
    if(maxx!=i)
    {
        swap(tree,maxx,i);
        heapify(tree,n,maxx);
    }
}
void build_heap(int tree[],int n)
{
    int last_node = n-1;
    int parent = (last_node-1)/2;
    int i;
    for(i = parent;i>=0;i--)
    {
        heapify(tree,n,i);
    }
}
void heap_sort(int tree[],int n)
{
    build_heap(tree,n);
    int i;
    for(i=n-1;i>=0;i--)
    {
        swap(tree,i,0);
        heapify(tree,i,0);
    }
}
int main()
{
    std::ios::sync_with_stdio(false);
    int tree[1000100];
    int n,m;
    cin>>n>>m;
    for(int q=0;q<n;q++)
        cin>>tree[q];
    heap_sort(tree,n);
    for(int i=n-1;i>=n-m;i--)
        cout<<tree[i]<<" ";
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值