剑指Offer : 数组中的逆序对

题目描述

在数组中的两个数字，如果前面一个数字大于后面的数字，则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007 

输入描述:

题目保证输入的数组中没有的相同的数字

数据范围：
对于%50的数据,size<=10^4
对于%75的数据,size<=10^5
对于%100的数据,size<=2*10^5




输入例子:
1,2,3,4,5,6,7,0


输出例子:

7

算法基于分治，归并排序的思想

代码如下：

typedef long long int64;

class Solution {
    public:

    int InversePairs(vector<int> data) {
        int size = data.size();
        vector<int> copy (size);
        int64 res = merge(data, copy, 0, data.size() - 1);
        return res % 1000000007;
    }

    private:

    int64 merge(vector<int> &sourceArr, vector<int> &tempArr, int start, int end) {
        int64 res = 0;
        if (start < end) {
            int middle = (start + end) >> 1;
            res += merge(sourceArr, tempArr, start, middle);
            res += merge(sourceArr, tempArr, middle + 1, end);
            res += merge_helper(sourceArr, tempArr, start, middle, end);
        }
        return res;
    }

    int64 merge_helper(vector<int> &sourceArr, vector<int> &tempArr, int start, int middle, int end) {
        int64 count = 0;
        int i = middle, j = end, k = end;
        //    i              j
        // start...... middle ....... end
        while (i >= start && j >= middle + 1) {
            if (sourceArr[i] > sourceArr[j]) {
                count += j - middle;
                tempArr[k--] = sourceArr[i--];
            } else {
                tempArr[k--] = sourceArr[j--];
            }
        }


        while (i >= start)
            tempArr[k--] = sourceArr[i--];


        while (j >= middle + 1)
            tempArr[k--] = sourceArr[j--];


        for (i = k + 1; i <= end; ++i) {
            sourceArr[i] = tempArr[i];
        }


        return count;
    }
};