数据结构实验之排序四：寻找大富翁

数据结构实验之排序四：寻找大富翁

Time Limit: 150ms   Memory limit: 512K  

题目描述

2015胡润全球财富榜调查显示，个人资产在1000万以上的高净值人群达到200万人，假设给出N个人的个人资产值，请你快速找出排前M位的大富翁。

输入

首先输入两个正整数N( N ≤ 10^6)和M(M ≤ 10)，其中N为总人数，M为需要找出的大富翁数目，接下来给出N个人的个人资产，以万元为单位，个人资产数字为正整数，数字间以空格分隔。

输出

一行数据，按降序输出资产排前M位的大富翁的个人资产值，数字间以空格分隔，行末不得有多余空格。

 

示例输入

6 3
12 6 56 23 188 60

示例输出

188 60 56

提示

请用堆排序完成。 

堆排序  链接： http://blog.csdn.net/xiaoxiaoxuewen/article/details/7570621/

       堆排序是利用堆的性质进行的一种选择排序。下面先讨论一下堆。

1.堆

  堆实际上是一棵完全二叉树，其任何一非叶节点满足性质：

  Key[i]<=key[2i+1]&&Key[i]<=key[2i+2]或者Key[i]>=Key[2i+1]&&key>=key[2i+2]

  即任何一非叶节点的关键字不大于或者不小于其左右孩子节点的关键字。

  堆分为大顶堆和小顶堆，满足Key[i]>=Key[2i+1]&&key>=key[2i+2]称为大顶堆，满足 Key[i]<=key[2i+1]&&Key[i]<=key[2i+2]称为小顶堆。由上述性质可知大顶堆的堆顶的关键字肯定是所有关键字中最大的，小顶堆的堆顶的关键字是所有关键字中最小的。

2.堆排序的思想

   利用大顶堆(小顶堆)堆顶记录的是最大关键字(最小关键字)这一特性，使得每次从无序中选择最大记录(最小记录)变得简单。

    其基本思想为(大顶堆)：

    1)将初始待排序关键字序列(R1,R2....Rn)构建成大顶堆，此堆为初始的无须区；

    2)将堆顶元素R[1]与最后一个元素R[n]交换，此时得到新的无序区(R1,R2,......Rn-1)和新的有序区(Rn),且满足R[1,2...n-1]<=R[n]; 

    3)由于交换后新的堆顶R[1]可能违反堆的性质，因此需要对当前无序区(R1,R2,......Rn-1)调整为新堆，然后再次将R[1]与无序区最后一个元素交换，得到新的无序区(R1,R2....Rn-2)和新的有序区(Rn-1,Rn)。不断重复此过程直到有序区的元素个数为n-1，则整个排序过程完成。

    操作过程如下：

     1)初始化堆：将R[1..n]构造为堆；

     2)将当前无序区的堆顶元素R[1]同该区间的最后一个记录交换，然后将新的无序区调整为新的堆。

    因此对于堆排序，最重要的两个操作就是构造初始堆和调整堆，其实构造初始堆事实上也是调整堆的过程，只不过构造初始堆是对所有的非叶节点都进行调整。

    下面举例说明：

     给定一个整形数组a[]={16,7,3,20,17,8}，对其进行堆排序。

    首先根据该数组元素构建一个完全二叉树，得到

 

 然后需要构造初始堆，则从最后一个非叶节点开始调整，调整过程如下：

20和16交换后导致16不满足堆的性质，因此需重新调整

这样就得到了初始堆。

即每次调整都是从父节点、左孩子节点、右孩子节点三者中选择最大者跟父节点进行交换(交换之后可能造成被交换的孩子节点不满足堆的性质，因此每次交换之后要重新对被交换的孩子节点进行调整)。有了初始堆之后就可以进行排序了。

此时3位于堆顶不满堆的性质，则需调整继续调整

 这样整个区间便已经有序了。

    从上述过程可知，堆排序其实也是一种选择排序，是一种树形选择排序。只不过直接选择排序中，为了从R[1...n]中选择最大记录，需比较n-1次，然后从R[1...n-2]中选择最大记录需比较n-2次。事实上这n-2次比较中有很多已经在前面的n-1次比较中已经做过，而树形选择排序恰好利用树形的特点保存了部分前面的比较结果，因此可以减少比较次数。对于n个关键字序列，最坏情况下每个节点需比较log2(n)次，因此其最坏情况下时间复杂度为nlogn。堆排序为不稳定排序，不适合记录较少的排序。

 刚开始没注意到内存和时间，提交后不是TLE就是MLE：

#include<stdio.h>
#define MaxN 1000000
int n,m,a[MaxN];
void sift(int l,int r)
{
    if(l<r)
    {
        int i=l,j=2*i,tmp=a[l];
        while(j<=r)
        {
            if(j<r&&a[j+1]>a[j])
                j++;
            if(a[i]<a[j])
            {
                a[i]=a[j];
                i=j;
                j=2*i;
            }
            else break;
        }
        a[i]=tmp;
    }
}
void HeapSort()
{
    int tmp;
    for(int i=n/2;i>0;i--)
        sift(i,n);
    for(int i=n;i>n-m;i--)
    {
         if(i==n)
            printf("%d",a[1]);
        else printf(" %d",a[1]);
        tmp=a[1];
        a[1]=a[i];
        a[i]=tmp;
        sift(1,i-1);
    }
}
int main()
{
    while(~scanf("%d %d",&n,&m))
    {
        for(int i=1; i<=n; i++)
            scanf("%d",&a[i]);
        HeapSort();
        printf("\n");
    }
    return 0;
}

后来发现M很小，so：

#include<stdio.h>
#define MaxN 12
int a[MaxN];
int n,m;
void Heapjust(int low,int high) //调整堆算法
{
    if(low<high)
    {
        int i,temp=a[low];
        for(i=2*low; i<=high; i*=2)
        {
            if(i<high&&a[i]>a[i+1])
                i++;
            if(temp<a[i])
                break;
            a[low]=a[i];
            low=i;
        }
        a[low]=temp;
    }
}

int main()
{
    int i,j,t;
    while(~scanf("%d %d",&n,&m))
    {
        for( i=1; i<=m; i++)
            scanf("%d",&a[i]);
        for(i=m/2;i>0;i--)
            Heapjust(i,m);      //构造初始小顶堆
        for(i=m+1;i<=n;i++)
        {
            scanf("%d",&t);
            if(a[1]>=t)              //a[1]是当期数组中最小的
                continue;
            a[1]=t;
            for(j=m/2;j>0;j--)  //不断更新小顶堆
                Heapjust(j,m);
        }
        for(i=m;i>0;i--)    //进行m-1次排序，每一趟元素减一
        {
            int tmp=a[1];       //将每次排序得到的最小值与当前区间的最后一个位置的元素交换
            a[1]=a[i];
            a[i]=tmp;
            Heapjust(1,i-1);
        }
       for(i=1;i<=m;i++)
        printf("%d%c",a[i],i==m?'\n':' ');
    }
    return 0;
}

PS：在38行代码时原本是“ if(a[i]>t) ”,第一次提交160ms，以后都是TLE，但是我发现好多人都是150ms甚至更底，优化一下，第一次150ms，第二次140ms，这OJ可真是。。。