实现原地堆排序（CPP）

堆是一种非常有用的数据结构，我们可以利用大顶堆或者小顶堆这种数据结构完成堆排序，但是这样会增加O（n）的辅助空间，其实，真正的堆排序是可以在原地进行的，我们称之为原地堆排序，今天我们就来实现一下原地堆排序吧。

原地堆排序主要分两个步骤：

1. 对数组进行堆化

2. 将最大元素与数组末端元素交换，然后对前n-1个元素继续堆化。

具体代码为：

#include <iostream>
#include <algorithm>

using namespace std;


template<typename T>
void __shiftDown(T arr[], int n, int k){

    while( 2*k+1 < n ){
        int j = 2*k+1;
        if( j+1 < n && arr[j+1] > arr[j] )
            j += 1;

        if( arr[k] >= arr[j] )break;

        swap( arr[k] , arr[j] );
        k = j;
    }
}

template<typename T>
void heapSort(T arr[], int n){
    // 首先对数组元素进行堆化
    // 还是对非叶子节点遍历实现shiftDown操作
    for( int i = (n-1)/2 ; i >= 0 ; i -- )
        __shiftDown(arr, n, i);

    // 循环使最大元素放置数组尾部，然后对剩余数据进行堆化
    for( int i = n-1; i > 0 ; i-- ){
        swap( arr[0] , arr[i] );
        __shiftDown(arr, i, 0);
    }
}

 

 

和直接插入排序类似，一次swap包含三次赋值操作，我们可以将shiftDown函数中的swap操作换成赋值操作。代码如下：

#include <iostream>
#include <algorithm>

using namespace std;

template<typename T>
void __shiftDown2(T arr[], int n, int k){

    T e = arr[k];
    while( 2*k+1 < n ){
        int j = 2*k+1;
        if( j+1 < n && arr[j+1] > arr[j] )
            j += 1;

        if( e >= arr[j] ) break;

        // swap操作换成了赋值操作，对程序进行了时间上的优化
        arr[k] = arr[j];
        k = j;
    }

    arr[k] = e;
}


template<typename T>
void heapSort(T arr[], int n){

    for( int i = (n-1)/2 ; i >= 0 ; i -- )
        __shiftDown2(arr, n, i);

    for( int i = n-1; i > 0 ; i-- ){
        swap( arr[0] , arr[i] );
        __shiftDown2(arr, i, 0);
    }
}