求数组中逆序对——分治算法

求数组中逆序对——分治算法

问题描述

见算法导论第三版5.2-5.

题目分析

该问题可以借助快速排序的主代码，在左右两侧子数组排序、子数组合并的过程中加入逆序对个数计算的过程。
基本过程如下：
1、如果数组计算区域仅包含一个元素，返回0；
2、否则，将数组分为左右两段[left,mid)与[mid,right)，对两个子数组进行排序，并计算左侧逆序对数目a和右侧逆序对数目b；
3、合并：计算横跨左右的逆序对数目c（该过程技巧位于while循环处：如果左侧数组元素array[ptrL]值大于右侧元素array[ptrR],那么对于所有ptrL<mid,有array[ptrL]>array[ptrR]），返回a+b+c；

代码

template <class T>
size_t InversionPairNumber_divide_and_conquer(std::vector<T> &array,
                                              size_t left, size_t right)
{
    //  [left,ight)
    if (left + 1 == right)
    {
        return 0;
    }

    size_t mid = left + (right - left) / 2;
    size_t a = InversionPairNumber_divide_and_conquer(array, left, mid);
    size_t b = InversionPairNumber_divide_and_conquer(array, mid, right);

    size_t ptrL = left, ptrR = mid;
    size_t c = 0;
    std::vector<T> sorted_array;
    while (ptrL < mid && ptrR < right)
    {
        if (array[ptrL] > array[ptrR])
        {
            sorted_array.push_back(array[ptrR++]);
            c += mid - ptrL;
        }
        else
        {
            sorted_array.push_back(array[ptrL++]);
        }
    }
    for (; ptrL < mid; sorted_array.push_back(array[ptrL++]))
        ;
    for (; ptrR < right; sorted_array.push_back(array[ptrR++]))
        ;
    for(size_t i =0;i<right-left;i++)
    {
        array[left+i] = sorted_array[i];
    }
    return a + b + c;
}

template <class T>
size_t Compute(std::vector<T> array, size_t left, size_t right)
{
    //隔绝引用对原array的修改
    size_t answer = InversionPairNumber_divide_and_conquer<T>(array,
                                                              0, array.size());
    return answer;
}

/* 使用示例
std::vector<int> array{-1,2, 1,0};
size_t answer = Compute<int>(array, 0, array.size());
*/

todo

网上有更快速的解法，挖坑待填。