数组中的逆序对(C++)

题目:

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

思路:

使用递归,将数组分为两部分,一直到分开的数组只包含一个元素的时候,分别使用两个指针,指向两个数组的最后一个元素,前一个数组的最后一个元素,如果前一个比后一个大,那就说明此时可以构成一个逆序对,然后把大的放到备用数组的后一位小的放到前一位,这样就实现了两个元素的排序。一步步向上递归,直到统计完所有逆序对,并且备用数组中为对原数组的排序。

代码实现:

class Solution {
public:
    int InversePairs(vector data) {
        int len = data.size();
        if(len <= 1)
            return 0;
        vector copy;
        for(int i = 0; i < len; i++)
            copy.push_back(data[i]);
        int result = InversePairsCore(data, copy, 0, len - 1);
        return result;
    }
    int InversePairsCore(vector &data, vector ©, int start, int end)
    {
        if(start == end)
        {
            copy[start] = data[start];
            return 0;
        }
        int mid = ((end - start) / 2);
        int left = InversePairsCore(copy, data, start, start + mid);
        int right = InversePairsCore(copy, data, start + mid + 1, end);
        int pre_end = start + mid;
        int post_end = end;
        int copy_idx = end;
        long long count = 0; // 测试用例中的数据较多,需要使用long long类型
        while(pre_end >= start && post_end >= start + mid + 1)
        {
            if(data[pre_end] > data[post_end]) // 前半字数组中元素大于后半子数组中最大元素时,逆序对数等于后一个数组的长度,并将最后大的元素复制到备用数组中
            {
                count = count + post_end - start - mid;
                copy[copy_idx--] = data[pre_end--];
            }
            else
            {
                copy[copy_idx--] = data[post_end--];
            }
        }
        /*---将其余的元素放到备用数组头---*/
        for(; pre_end >= start; pre_end--)
            copy[copy_idx--] = data[pre_end];
        for(; post_end >= start + mid + 1; post_end--)
            copy[copy_idx--] = data[post_end];
        /*---返回计数对1000000007取余的结果---*/
        return (count + left + right) % 1000000007;
    }
};

 

你可能感兴趣的:(C++)