c++ - 使用指向数组的指针进行合并排序

Question

我正在尝试使用 STL 库创建合并排序算法，但遇到了一些问题。下面是我正在使用的代码

template <typename Item, typename SizeType>
void merge_sort(Item array[], SizeType size){
    size_t n1; //Size of the first subarray
    size_t n2; //Size of the second subarray

    if(size > 1){
        //Compute the size of the subarrays
        n1 = size/2;
        n2 = size - n1;

        //create the temp array.
        int* n1Temp = new int[n1];
        int* n2Temp = new int[n2];
        int i;
        for(i = 0; i < n1; i++)
            n1Temp[i] = array[i];
        for(i = 0; i < n2; i++)
            n2Temp[i] = array[i + n1];

        //recursive calls
        merge_sort(n1Temp, n1);//sort from array[0] through array[n1 - 1] 
        merge_sort(n2Temp, n2);//sort from array[n1] to the end

        //Merge the two sorted halves.
        vector<int> v(array, array + size);
        merge(n1Temp, n1Temp + n1, n2Temp, n2Temp + n2, v.begin());     
        copy(v.begin(), v.end(), array);//copy the vector back to the array

        delete[] n1Temp;
        delete[] n2Temp;
    }
}

代码排序很好，但问题是它像 O(n^2) 算法而不是 O(n\log n)，这是由于在每次合并排序调用中创建 vector (我认为).我尝试删除 vector 并仅在合并函数中使用一个数组，如下所示

    //Merge the two sorted halves.
    int* finalArray = new int[n1 + n2];
    merge(n1Temp, n1Temp + n1, n2Temp, n2Temp + n2, begin(finalArray)); 
    array = finalArray;

但这只会给我带来错误。我可以做些什么来挽救我的合并排序算法吗？

Answer 1

正如 Vaughn 和 user93353 所指出的，您应该能够在每个合并点直接合并到目标数组中。但是您仍然可以使用 std::vector<> 来大大简化此操作。

此外，您的临时数组是直接类型“int”，我相当确定它是模板参数 Item 的类型。我不确定 SizeType 参数的用途，但我保留了它以防您有特殊想法。不管是什么，最好与 size_t 兼容:

template <typename Item, typename SizeType>
void merge_sort(Item array[], SizeType size)
{
    if(size > 1)
    {
        //Compute the size of the subarrays
        size_t n1 = size/2;

        //create the temp array
        std::vector<Item> n1Temp(array, array+n1);
        std::vector<Item> n2Temp(array+n1, array+size);

        //recursive calls
        merge_sort(&n1Temp[0], n1);       //sort array[0] through array[n1-1]
        merge_sort(&n2Temp[0], size-n1);  //sort array[n1] through array[size-1]

        // merge the sorted halves
        std::merge(n1Temp.begin(), n1Temp.end(),
                   n2Temp.begin(), n2Temp.end(), array);
    }
}

上述技术通过复制自上而下拆分子序列，然后将拆分拷贝就地合并到原始数组中。您可以通过对原始数组进行拆分，然后合并到临时空间并在之后复制，将此算法减少一个子列表分配时间(但不会减少空间)，我认为 你一开始想做的事:

template <typename Item>
void merge_sort(Item ar[], size_t n)
{
    if (n > 1)
    {
        // Compute the size of the subarrays
        size_t n1 = n/2;

        // invoke recursion on the submerges
        merge_sort(ar, n1);      //sort array[0] through array[n1-1]
        merge_sort(ar+n1, n-n1); //sort array[n1] through array[size-1]

        // create merge-buffer
        std::vector<Item> mrg;
        std::merge(ar, ar+n1, ar+n1, ar+n, back_inserter(mrg));
        std::copy(mrg.begin(), mrg.end(), ar);
    }
}

---

基于迭代器的通用解决方案

对于允许更大灵 active 的通用解决方案，您可以根据迭代器而不是项目指针来定义合并排序。它变得有点多毛，但好处是非常 std-lib-ish。

template <typename Iterator>
void merge_sort(Iterator first, Iterator last)
{
    typedef typename std::iterator_traits<Iterator>::value_type value_type;
    typedef typename std::iterator_traits<Iterator>::difference_type difference_type;

    difference_type n = std::distance(first, last)/2;
    if (n == 0)
        return;

    // invoke recursion on the submerges
    merge_sort(first, first + n);
    merge_sort(first + n, last);

    // create merge-buffer
    std::vector<value_type> mrg(std::distance(first, last));
    std::merge(first, first+n, first+n, last, mrg.begin());
    std::copy(mrg.begin(), mrg.end(), first);
}

---

最后，如果您发现自己要对大量固定长度的 C 数组进行排序，您可能会发现以下内容很有用(它使用上面的通用迭代器解决方案):

// front-loader for C arrays
template<typename Item, size_t N>
void merge_sort(Item (&ar)[N])
{
    merge_sort(std::begin(ar), std::end(ar));
}

它使下面的代码相当方便:

int arr[1024];
... fill arr ...
merge_sort(arr);