堆排序

堆排序

// 将待排序的最后一个元素current插入大根堆中
void insert_heap(int *arr, int current, int low, int high)
{
    //记录low的孩子结点中较大元素的下标，初始化为左孩子
    int large = 2*low + 1;
    while(large <= high)
    {
        //右孩子较大
        if(large < high && arr[large] < arr[large+1])
            large++;
        //当前元素大于所有孩子
        if(current >= arr[large])
            break;
        else//当前元素小于左右孩子中较大的一个
        {
            arr[low] = arr[large];
            low = large;
            large = 2*low + 1;
        }
    }
    //将目标元素插入合适的位置
    arr[low] = current;
}

//建堆
void build_heap(int *arr, int n)
{
    int low, current;
    //从最后一个非叶结点开始
    for(low = n/2 - 1; low >= 0; low--)
    {
        current = arr[low];
        insert_heap(arr,current,low,n-1);
    }
}

//堆排序，从后往前，每次将堆头最大的元素放到堆尾，同时将堆尾元素插入到堆中
void hearSort(int *arr, int n)
{
    int current;//暂存待插入的元素
    int last_unsorted;//未排序的最后一个元素
    build_heap(arr,n);//建堆

    for(last_unsorted = n-1; last_unsorted > 0; last_unsorted--)
    {
        current = arr[last_unsorted];//暂存堆尾元素
        arr[last_unsorted] = arr[0];//堆头元素放在堆尾
        insert_heap(arr,current,0,last_unsorted-1);//未排序的堆尾元素插入堆
    }
}