35. 数组中的逆序对

题目描述

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

思路:归并排序,同时计算逆序对的个数。

class Solution {
public:
    int InversePairs(vector<int> data) {
        int len = data.size();
        if(len <= 0)
            return 0;
        vector<int> copy;
        copy = vector<int> (data.begin(), data.end());
        long long count = MergeSort_Count(data, copy, 0, len - 1);
        return count % 1000000007;
    }
    long long MergeSort_Count(vector<int> &data, vector<int> &copy, int start, int end)
    {
        if(start == end)
            return 0;
        int len = (end - start)/2;
        long long left = MergeSort_Count(copy, data, start, start + len);
        long long right = MergeSort_Count(copy, data, start + len + 1, end);
        int i = start + len;//前半段最后一个数字的下标
        int j = end;//后半段最后一个数字的下标
        int copyindex = end;
        long long count = 0;
        while(i >= start && j >= start + len + 1)
        {
            if(data[i] > data[j])
            {
                copy[copyindex --] = data[i --];
                count = count + j - start - len;//j - start - len指逆序对的个数是从后半段第一个数字到当前数字的所有个数
            }
            else
                copy[copyindex --] = data[j --];
        }
        for(; i >= start; i --)//这两个for循环的作用,你自己单步调试一遍就知道了:如果第一个for循环起作用,那么第二个就不运行,反之,如果第二个for循环不运行,那么第一个就会运行,你自己跑一下就知道了。
            copy[copyindex --] = data[i];
        for(; j >= start + len + 1; j --)
            copy[copyindex --] = data[j];
        return count + left + right;
    }
};

猜你喜欢

转载自blog.csdn.net/weixin_39605679/article/details/81224301