PTA 堆排序 函数题 c++

本题要求实现一个函数，对n个数据进行排序（从小到大），只要求得到前m个最大值即可。

函数接口定义：

void sort(int a[],int n,int m)；

其中 a 、 n 、m都是用户传入的参数。 n 是数据元素个数； m 是统计个数。

裁判测试程序样例：

#include <stdio.h>

void sort(int a[],int n,int m);

int main()
{
    int i,n,m,*a;
    scanf("%d%d",&n,&m);
    a=new int[n+1];
    for(i=1;i<=n;i++)
        scanf("%d",&a[i]);
    sort(a,n,m);
    for(i=n;i>n-m;i--)
        printf("%d ",a[i]);
    return 0;
}
/* 请在这里填写答案 */

输入样例：

10 3
12 65 45 67 32 83 23 54 29 96

输出样例：

96 83 67 

下面是我的解法+注释

//堆的性质必须满足两点：符合完全二叉树、父节点大于（小于）子节点
//一个顶堆就包含三个节点：父节点、左子节点、右子节点
//在裁判测试中它是从后往前输入，说明该堆排序是升序，则所有父节点大于子节点
//heap函数用来维护堆的性质，i代表当前需要维护的节点下标,n代表一共有多少个节点
void heap(int a[],int i,int n)
{
    int largest=i;      //父节点
    int l=i*2;          //左子节点
    int r=i*2+1;        //右子节点
    if(l<=n&&a[largest]<a[l])   //必须小于n
        largest=l;
    if(r<=n&&a[largest]<a[r])
        largest=r;
    if(largest!=i)
    {
        int t;
        t=a[largest];a[largest]=a[i];a[i]=t;   //使父亲节点永远大于子节点
        heap(a,largest,n);
    }
}
void sort(int a[],int n,int m)
{
    int i;
    for(i=n/2;i>=1;i--)   //从n/2到1的调用顺序可以确保所有非叶子节点都被处理到，从而构建出一个完整的堆
        heap(a,i,n);
    for(i=n;i>1;i--)
    {
        int t;
        t=a[i];a[i]=a[1];a[1]=t;//交换之后，a[i]已经不属于该堆，所以需要删掉总数
                                //相当于a[i]已经在该数组上排序好了
        heap(a,1,i-1);  //继续维护堆的性质
    }
}