堆排序C++算法

/*
堆排序
*/
#include <iostream>
#define MAX 11
//initialization
int a[MAX] = {3,6,2,7,9,1,56,78,23,55,11}; 


int heapBuild(int *a,int i,int n)
{
//如果heap位置小于1，就表示结束
if(i < 1)
return 0;
int max;
//2i > n表示超出范围，没有必要进一步调整
if(2*i > n)
return 0;
//只有一个节点
if(2*i == n)
{
if(a[2*i-1] > a[i-1])
{
max = a[2*i-1];
a[2*i-1] = a[i-1];
a[i-1] = max;
}
return 0;
}
//进行调整
if(2*i < n)
{
if(a[2*i] < a[2*i-1])
{
if(a[i-1] < a[2*i-1])
{
max = a[2*i-1];
a[2*i-1] = a[i-1];
a[i-1] = max;
heapBuild(a,2*i,n);//调整了一个，可能影响了子树，调整子树
}


}
else{ 
if(a[2*i-1] < a[2*i])
if(a[i-1] < a[2*i])
{
max = a[2*i];
a[2*i] = a[i-1];
a[i-1] = max;
heapBuild(a,2*i+1,n);
}
}
}
heapBuild(a,i-1,n);
return 1;
}


void print(int *a,int n)
{
int i;
for(i = 0; i < n;i++)
{
std::cout<<a[i]<<std::endl;
}
}
//排序
void sortHeap(int *a,int n)
{
int max;
int len = n;
while(len > 0)
{
//将第一个与最后一个交换。
max = a[0];
a[0] = a[len-1];
a[len-1] = max;
len = len -1;
int i = len/2;
heapBuild(a,1,len);
}
}
int main()
{
int i = MAX/2;
heapBuild(a,i,MAX);
sortHeap(a,MAX);
print(a,MAX);
return 1;
}