题目:
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007
思路:
使用递归,将数组分为两部分,一直到分开的数组只包含一个元素的时候,分别使用两个指针,指向两个数组的最后一个元素,前一个数组的最后一个元素,如果前一个比后一个大,那就说明此时可以构成一个逆序对,然后把大的放到备用数组的后一位小的放到前一位,这样就实现了两个元素的排序。一步步向上递归,直到统计完所有逆序对,并且备用数组中为对原数组的排序。
代码实现:
class Solution {
public:
int InversePairs(vector<int> data) {
int len = data.size();
if(len <= 1)
return 0;
vector<int> 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<int> &data, vector<int> ©, 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;
}
};