[剑指Offer] 35.数组中的逆序对

题目描述

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

【思路】看到这样的题目，最简单的想法就是遍历每一个元素，让其与后面的元素对比，如果大于则count++，但是这样的时间复杂度是o(n²)，根据题目给出的数据量，很明显超时。因此想到了用归并排序的思想。

class Solution {
public:
    void MergeCount(vector<int>& data, int begin, int mid, int end, int& count)
    {
        int i = begin, j = mid;
        int temp[end - begin],  k = 0;
        while(i < mid && j < end)
        {
            if(data[i] < data[j])
            {
                temp[k++] = data[i++];
            }
            else
            {
                count += mid - i;
                if(count > 1000000007) count %= 1000000007;
                temp[k++] = data[j++];
            }
        }
        while(i < mid) temp[k++] = data[i++];
        while(j < end) temp[k++] = data[j++];
        for(int s = 0; s < end - begin; ++s) data[s + begin] = temp[s];
    }
    void MergeSort(vector<int>& data, int begin, int end, int& count)
    {
        if(begin + 1 == end) return;
        int mid = (begin + end) / 2;
        MergeSort(data, begin, mid,count);
        MergeSort(data, mid, end, count);
        MergeCount(data, begin, mid, end, count);
    }
    int InversePairs(vector<int> data)
    {
        int n = static_cast<int>(data.size());
        if(n == 0 || n == 1) return 0;
        int count = 0;
        MergeSort(data,0, n, count);
        return count;
    }
};