题目描述
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007。
class Solution {
public:
//合并两个已排序子数组,并统计逆序对,返回合并后排好序的数组
vector<int> merge_count(const vector<int>& vec1,const vector<int>& vec2,long long &count)
{
vector<int> res(vec1.size()+vec2.size(),0);
int i=vec1.size()-1,j=vec2.size()-1;
int index=vec1.size()+vec2.size()-1;
while(i>=0&&j>=0)
{
if(vec1[i]>vec2[j])
{
count+=j+1;
res[index--]=vec1[i];
--i;
}
else
{
res[index--]=vec2[j];
--j;
}
}
if(i<0)
{
while(j>=0)
{
res[index--]=vec2[j--];
}
}
else
{
while(i>=0)
{
res[index--]=vec1[i--];
}
}
count%=1000000007;
return res;
}
int InversePairs(vector<int> data) {
int mark_size=1;
long long count=0;
//用来存放需要合并的数组,牺牲了大量空间,但更为容易理解
vector<vector<int>> vec;
while(mark_size<data.size())
{
if(mark_size==1)
{
int i=0;
for(i=0;i+1<data.size();i+=2)
{
vec.push_back(merge_count(vector<int>(1,data[i]),\
vector<int>(1,data[i+1]),count));
}
if(i<data.size())
{
vec.push_back(vector<int>(1,data[i]));
}
}
else
{
vector<vector<int>> mark_vec;
int i=0;
for(;i+1<vec.size();i+=2)
{
mark_vec.push_back(merge_count(vec[i],vec[i+1],count));
}
if(i<vec.size())
{
mark_vec.push_back(vec[i]);
}
vec=mark_vec;
}
mark_size*=2;
}
return count;
}
};