您的位置:首页 > 其它

hdu1425 sort

2012-09-01 11:51 218 查看
用堆排序写的,可能优化的不是很好,加上输入外挂才没有超时

code:

//从今天开始不再用宏定义
//采用松散的代码写法~~
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <vector>
#include <cstring>
using namespace std;
void swap(int &a1,int &a2)
{
int tmp=a2;
a2=a1;
a1=tmp;
}
void max_heapfy(int A[],int i,int heap_size)
{
int l=i<<1;
int r=i<<1|1;
int largest;

if(l<=heap_size && A[l]>A[i])
{
largest=l;
}else{
largest=i;
}

if(r<=heap_size && A[r]>A[largest])
{
largest=r;
}

if(largest!=i)
{
swap(A[i],A[largest]);
max_heapfy(A,largest,heap_size);
}
}
void build_max_heap(int A[],int heap_size)
{
int i;
for(i=heap_size/2;i>=1;i--)
{
max_heapfy(A,i,heap_size);
}
}
void heap_sort(int A[],int len)
{
int i;
build_max_heap(A,len);
for(i=len;i>=2;i--)
{
swap(A[1],A[len]);
len--;
max_heapfy(A,1,len);
}
}
inline bool scan_d(int &num)
{
char in;bool IsN=false;
in=getchar();
if(in==EOF) return false;
while(in!='-'&&(in<'0'||in>'9')) in=getchar();
if(in=='-')
{
IsN=true;num=0;
}
else num=in-'0';
while(in=getchar(),in>='0'&&in<='9')
{
num*=10,num+=in-'0';
}
if(IsN) num=-num;
return true;
}
int dat[1000010],n,m;
int main()
{
int i;

while(~scanf("%d%d",&n,&m)){
for(i=1;i<=n;i++)
{
scan_d(dat[i]);
}
heap_sort(dat,n);
//        for(i=n;i>=1;i--)
//        {
//            printf("%d ",dat[i]);
//        }
for(i=n;i>n-m+1;i--)
{
printf("%d ",dat[i]);
}
printf("%d\n",dat[n-m+1]);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: