hdu 1425 堆排序

sort

Time Limit: 6000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 24698    Accepted Submission(s): 7467

Problem Description

给你n个整数，请按从大到小的顺序输出其中前m大的数。

 

Input

每组测试数据有两行，第一行有两个数n,m(0<n,m<1000000)，第二行包含n个各不相同，且都处于区间[-500000,500000]的整数。

 

Output

对每组测试数据按从大到小的顺序输出前m大的数。

 

Sample Input

  
  

   
   5 3
3 -35 92 213 -644
  
  

 

Sample Output

  
  

   
   213 92 3

  
  
  
  


  
  

#include<iostream>
#include<cstdio>
#include<string.h>
using namespace std;

int a[1000005];

void heapjust(int w,int n)
{
    int temp=a[w];
    int lchild=2*w+1;
    int p=w;
    while(lchild<n)
    {
        if( (lchild<n-1)&&a[lchild]>a[lchild+1] )
         lchild++;
        if(temp>a[lchild])
        {
            a[p]=a[lchild];
            p=lchild;
            lchild=2*p+1;
        }
        else break;
    }
    a[p]=temp;
}
void heapsort(int n)
{
    int i,j,k;
    for(i=(n-1)/2;i>=0;i--)
      heapjust(i,n);

    for(j=n-1;j>0;j--)
    {
        k=a[0];
        a[0]=a[j];
        a[j]=k;
        heapjust(0,j);
    }

}
int main()
{
    int i,n,m;
    while(~scanf("%d%d",&n,&m))
    {
        for(i=0;i<n;i++)
          scanf("%d",&a[i]);
        heapsort(n);
        if(m>n) m=n;
        printf("%d",a[0]);
        for(i=1;i<m;i++)
         printf(" %d",a[i]);
         printf("\n");
    }
    return 0;
}