【剑指offer】数组中的逆序对（排序）

题目描述

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

输入描述:

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

数据范围：

对于%50的数据,size<=10^4

对于%75的数据,size<=10^5

对于%100的数据,size<=2*10^5

示例1
输入

1,2,3,4,5,6,7,0

输出

7

思路

采用暴力的方法时间复杂度为O（n^2）
考虑采用归并排序的方法，将时间复杂度降到O（nlogn）。统计相邻的的两个子区间的逆序对数，并将两个子区间排序。

代码

class Solution {
public:
    int InversePairs(vector<int> data) {
        if(data.size() <= 1)
            return 0;
        int count = 0;
        vector<int> copy(data); // 初始化
        InversePairsCore(data, copy, 0, data.size()-1,count);
        return count;
    }

    // 归并
    void InversePairsCore(vector<int> &data, vector<int> &copy, int start, int end, int &count){
        if(start>=end){
            return;
        }
        int mid = (start + end) /2;
        InversePairsCore(data,copy,start,mid,count);
        InversePairsCore(data,copy,mid+1,end,count);

        int copyIndex = end; // 将较大数字从后往前复制到辅助数组
        int i = mid;    // 前半段最后一个元素下标
        int j = end;    // 后半段最后一个元素下标
        while (i>=start && j>=mid+1) {
            if (data[j] < data[i]){     // 逆序
                count += j - mid;
                count %= 1000000007;    // 取模防止逆序对溢出
                copy[copyIndex--] = data[i--];
            } else{
                copy[copyIndex--] = data[j--];
            }
        }

        while (i>=start)
            copy[copyIndex--] = data[i--];

        while (j>=mid+1)
            copy[copyIndex--] = data[j--];

        for (int k = start; k <= end; ++k)
            data[k] = copy[k];

    }
};