hdu 1280 用堆

前m大的数

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 2922    Accepted Submission(s): 1013

Problem Description

还记得Gardon给小希布置的那个作业么？（上次比赛的1005）其实小希已经找回了原来的那张数表，现在她想确认一下她的答案是否正确，但是整个的答案是很庞大的表，小希只想让你把答案中最大的M个数告诉她就可以了。 
给定一个包含N(N<=3000)个正整数的序列，每个数不超过5000，对它们两两相加得到的N*(N-1)/2个和，求出其中前M大的数(M<=1000)并按从大到小的顺序排列。

 

Input

输入可能包含多组数据，其中每组数据包括两行： 
第一行两个数N和M， 
第二行N个数，表示该序列。

 

Output

对于输入的每组数据，输出M个数，表示结果。输出应当按照从大到小的顺序排列。

 

Sample Input

4 4
1 2 3 4
4 5
5 3 6 4

 

Sample Output

7 6 5 5
11 10 9 9 8

 

Author

Gardon

 

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
//#include<debug.h>
const int inf=4498501;
int M;
struct node
{
 int sum;
}C[inf];
int A[3000],B[3000];

/*
int cmp(const void *a,const void *b)
{
    return (*(node *)a).sum < (*(node *)b).sum ? 1:-1;
}
*/
int swap(int &x,int &y)
{
    int temp;
     temp=x;
     x=y;
     y=temp;
     return 0;
}
void down(int n,int i)
{
   int flag=1;
   if((2*i)<=n)
   {  
      while(flag&& (i=2*i)<=n)
      {
      if(i+1<=n&&C[i+1].sum>C[i].sum)
      i++;
      if(C[i/2].sum<C[i].sum)
      {
      
      swap(C[i/2].sum,C[i].sum);
      
      }
      else 
      flag=0;
    
     } 
}
  
}    
void makeheap(int n )
{
  C[n].sum=C[0].sum;
  int i;
  for(i=n/2;i>=1;i--)
  down(n,i);
 // printf("**:%d\n",C[1]);
  
} 

void head_sort(int n)
{
    int i;
    makeheap(n);
    for(i=n;i>=n-M+1;i--)
    {
     swap(C[1].sum,C[i].sum);
   
     down(i-1,1);
    
    }
}   
    
  
int main( )
{
 int N;
 //Debug();
 while(scanf("%d%d",&N,&M)!=EOF)
 {
  int P,k=0;
 // P=N*(N-1)/2;
  
  int i,j;
  for(i=0;i<N;i++)
  {
   scanf("%d",&A[i]);
   //B[i]=A[i];
  }
  for(i=0;i<N;i++)
    for(j=i+1;j<N;j++)
    C[k++].sum=A[i]+A[j];
  
  
   head_sort(k);
   for(i=k;i>=k-M+1;i--)
   printf(i==k-M+1?"%d":"%d ",C[i].sum);
   printf("\n");
 }
 return 0;
}

用堆写的，代码直的很长，我用qsort函数写时，超时了，悲剧，，，，。。估计是排序时，把所有

的都排了，等下再人优化下，用qsort写的。。也还可以用sort函数。。