首页 > 代码库 > 算法导论第六章 堆排序

算法导论第六章 堆排序

建堆的时间复杂度是O(n),堆排序的时间复杂度是O(NLogN),具体算法如下所示:

#include <iostream>
using namespace std;
void swap(int &i,int &j)
{
    int temp=i;
    i=j;
    j=temp;
}
void shiftDown(int *A, int start,int len)
{
    int temp=A[start];
    int i=start;
    int j=2*i+1;
    while (j<len)
    {
        if (j+1<len && A[j+1]>A[j])
        {
            j++;
        }
        if(A[start]>A[j])
        {
            break;
        }else
        {
            A[i]=A[j];
            i=j;
            j=2*j+1;
        }
    }
    A[i]=temp;
}
void createHeap(int *A,int len)
{
    for(int i=(len-2)/2;i>=0;i--)
    {
        shiftDown(A, i,len);
    }
}
void heapSort(int *A,int len)
{
    createHeap(A,len);
    for (int i=0; i<len; i++)
    {
        swap(A[0],A[len-i-1]);
        shiftDown(A,0,len-i-1);
    }
}
int main(int argc, const char * argv[])
{
    //int A[]={100,11,43,65};
    //int A[]={56,3,5,68,100,32};
    int A[]={68,100,32};

    heapSort(A,sizeof(A)/sizeof(int));
    for(int i=0;i<sizeof(A)/sizeof(int);i++)
    {
       cout<<A[i]<<"   ";
    }
    cout <<endl;
    return 0;
}