题目描述
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数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> ©, 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;
}
};