牛客网
题目描述:在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007
class Solution {
public:
int nixu_num;
int InversePairs(vector<int> data) {
nixu_num = 0;
vector<int> ans = mergesort(data);
return (nixu_num%1000000007);
}
vector<int> mergesort(vector<int> data){
if(data.size()==0){
return {};
}
if(data.size()==1){
return data;
}
auto b = data.begin(),e = data.end();
vector<int> left = mergesort(vector<int>(b,b+(e-b)/2));
vector<int> right = mergesort(vector<int>(b+(e-b)/2,e));
vector<int> ans;
auto l = left.begin(),r = right.begin();
for(int i = 0;i<(left.size()+right.size());i++){
if(l != left.end()&&(r == right.end()||*l <= *r)){
ans.push_back(*l);
l++;
continue;
}
if(r != right.end()&&(l == left.end()||*l > *r)){
ans.push_back(*r);
r++;
nixu_num+=(left.end()-l);
nixu_num=nixu_num%1000000007;
continue;
}
}
return ans;
}
};
想着用哈希或者其他方法,一直没找到复杂度在O(nlogn)之下的方法,还是果断用归并吧。
计算逆序对时,直接计算左侧数组指针范围就可以了,另外这道题目的几个测试用例返回的数值比较大,不处理的话会不通过,需要在函数内部进行取余操作。