堆排序C++实现

// 堆排序.cpp : Defines the entry point for the console application.
//时间复杂度为nlgn
//建立最大堆

#include "stdafx.h"
#include<iostream>
using namespace std;
int A[100];
//
//保持堆得性质
//a 为待排序数组，sum是待排序元素个数 i是父元素序号
void keep_heapify(int *a,int sum,int i)
{
int left = 2*i;
int right = 2*i+1;
int largest = i;
if(left<=sum&&a[left]>a[i])
largest = left;
if(right<=sum&&a[right]>a[largest])
largest = right;
if(largest!=i)
{
int temp = a[i];
a[i] = a[largest];
a[largest] = temp;
keep_heapify(a,sum,largest);
}
}
//创建大堆
void create_heap(int *a,int sum)
{
int i;
for(i=sum/2;i>=1;i--)
{
keep_heapify(a,sum,i);
}
}

//堆排序
//首先建立大堆，然后交换堆顶和堆底的元素，待排序元素总数减一
//调用keep_heapify()保持堆性质，直到排完
void heap_sort(int *a,int sum)
{
create_heap(a,sum);
int j;
int num = sum;
for(j=sum;j>=2;j--)
{
num--;
int temp = a[1];
a[1]=a[j];
a[j]=temp;
keep_heapify(a,num,1);
}
}

int _tmain(int argc, _TCHAR* argv[])
{
int cases;
cout<<"请输入案例个数："<<endl;
cin>>cases;
while(cases--)
{
cout<<"请输入需要排序元素的个数："<<endl;
int n;
cin>>n;
int k;
cout<<"请输入需要排序的元素："<<endl;
for(k=1;k<=n;k++)
cin>>A[k];
cout<<"排序前："<<endl;
for(k=1;k<=n;k++)
cout<<A[k]<<" ";
cout<<endl;
heap_sort(A,n);
cout<<"堆排序后："<<endl;
for(k=1;k<=n;k++)
cout<<A[k]<<" ";
cout<<endl;
}
system("pause");
return 0;
}