swust oj 1015

堆排序算法

1000(ms)

10000(kb)

2631 / 5595

编写程序堆排序算法。按照从小到大的顺序进行排序，测试数据为整数。

输入

第一行是待排序数据元素的个数； 第二行是待排序的数据元素。（提示：用小根堆）

输出

一趟堆排序的结果。

样例输入

10
50 36 41 19 23 4 20 18 12 22

样例输出

4 12 20 18 22 41 50 36 19 23

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdio>
using namespace std;

void sift(int arr[],int low,int high)
{
    int i=low,j=2*i;
    int tmp=arr[i];
    while(j<=high)
    {
        if(j<high&&arr[j]>arr[j+1])  //小的上浮，大的下沉
            j++;
        if(tmp>arr[j])
        {
            arr[i]=arr[j];
            i=j;
            j=2*i;
        }
        else
            break;
    }
    arr[i]=tmp;
}

void Heapsort(int arr[],int n)
{
    for(int i=n/2;i>=1;i--)  //建立初始堆，第一趟
        sift(arr,i,n);
}

int main()
{
    int n;
    int arr[1000];
    cin>>n;
    for(int i=1;i<=n;i++)
        cin>>arr[i];
    Heapsort(arr,n);
    for(int i=1;i<=n;i++)
        cout<<arr[i]<<" ";
    return 0;
}