在练习题目过程中,需要用到排序。C++中有专门的排序函数qsort,采用快速排序算法实现。但是在待排序序列为有序序列时,快速排序算法复杂度变为O(n)。排序算法中堆排序算法在任何情况下复杂度都为O(nlogn) ,只需要O(1)的辅助空间,因此是比较优秀的排序算法。处于学习算法的目的,写一个采用堆排序算法的函数,函数原型仿照qsort()。
Code:
void heapSort(void * arr, int nElem, int nsize, int (*f)(const void *, const void *))
{
int i;
void *temp = malloc(nsize);
for(i=nElem>>1;i>0;i--)
sift(arr,i-1,nElem-1,nsize,f,temp/* 重复利用动态分配内存*/); //arr[i-1..nElem-1]
for(i=nElem-1;i>=1;i--)
{
memcpy(temp,((char *)arr)+i*nsize,nsize);
memcpy(((char *)arr)+i*nsize,((char *)arr),nsize);
memcpy(((char *)arr),temp,nsize);
sift(arr,0,i-1,nsize,f,temp);
}
free(temp);
}
static void sift(void *arr,int left, int right, int nsize,int (*f)(const void *,const void *),void *temp)
{
/*
* arr[left+1..right]已经满足堆的性质,
* 加入arr[left]使得arr[left..right]满足堆的性质
*/
int i = left;
int j = (i<<1)+1;
bool finished = false;
memcpy(temp,((char *)arr)+left*nsize,nsize);
while(j<=right&&!finished)
{
if(j<right&&f((void *)(((char *)arr)+j*nsize),(void *)(((char *)arr)+(j+1)*nsize))>0)
j++;
if(f(temp,(void *)(((char *)arr)+j*nsize))<=0)
finished = true;
else
{
memcpy(((char *)arr)+i*nsize,((char *)arr)+j*nsize,nsize);
i = j;
j<<=1;
j++;
}
}
memcpy(((char *)arr)+i*nsize,temp,nsize);
}
static int compare(const void *a, const void *b)
{
/*
* a>b return >0
* a=b return 0
* a<b return <0
*/
}
参数类型和说明和qsort一样。直接调用heapSort()函数即可实现排序。
如果有需要,我还会实现其它排序算法。
请大家提出宝贵意见!谢谢!