根据算法导论,堆排序讲解,写出两种解决方法。
#include <stdio.h>
#define LEFT(i) 2*i
#define RIGHT(i) 2*i+1
void swap(int *a,int *b)//交换数字
{
int temp = *a;
*a = *b;
*b = temp;
}
/*维护堆的性质*/
void max_heapify(int A[],int i,int n)//n表示堆的大小,i表示从第i个数字进行维护堆的性质,0不算
{
int l = LEFT(i);
int r = RIGHT(i);
int largest;
if(l<=n&&A[l]>A[i]){
largest = l;
}
else
largest = i;
if(r<=n&&A[r]>A[largest]){
largest = r;
}
if(i!=largest){
swap(A+i,A+largest);
max_heapify(A,largest,n);
}
}
/*建堆*/
void build_heap(int A[],int n)
{
for(int i=n/2;i>0;i--){
max_heapify(A,i,n);
}
}
/*堆排序*/
void heap_sort(int A[],int n)
{
build_heap(A,n);
for(int i=n;i>0;i--){
swap(A+1,A+i);
n--;
max_heapify(A,1,n);
}
}
int main()
{
int A[] = {0,-12,5,55,5,4,3,2,1};
heap_sort(A,8);
for(int i=1;i<9;i++){
printf("%d ",A[i]);
}
return 0;
}
第二种:
由于第一份代码把第一个数字即0号数字排除在外,所以现在解决这种不足
#include <stdio.h>
#define LEFT(i) 2*i+1//原来只是按照书上的思路定义的左孩子和右孩子的序号关系,但是和数组从0开始不同,所以这样就对啦
#define RIGHT(i) 2*i+2
void swap(int *a,int *b)
{
int temp = *a;
*a = *b;
*b = temp;
}
void max_heapify(int A[],int i,int n)
int l = LEFT(i);
int r = RIGHT(i);
int largest;
if(l<n&&A[l]>A[i]){
largest = l;
}
else
largest = i;
if(r<n&&A[r]>A[largest]){
largest = r;
}
if(i!=largest){
swap(A+i,A+largest);
max_heapify(A,largest,n);
}
}
/*??o???*/
void build_heap(int A[],int n)
{
int i;
for(i=(n-1)/2;i>0;i--){
max_heapify(A,i,n);
}
}
/*?????¡¯?o*/
void heap_sort(int A[],int n)
{
build_heap(A,n);
for(int i=n;i>0;i--){
swap(A,A+i-1);
n--;
max_heapify(A,0,n);
}
}
int main()
{
int A[] = {56,-12,5,55,5,4,3,2,1};
heap_sort(A,9);
for(int i=0;i<9;i++){
printf("%d ",A[i]);
}
return 0;
}
OK!!!