方法1:归并排序的思想
class Solution {
public:
int merge(vector<int>& nums,int l,int mid,int r)
{
vector<int> help(r-l+1);
for(int i=l;i<=r;i++)
{
help[i-l]=nums[i];
}
//在归并排序之前,统计翻转对的数量
int res=0;
int i=l;
int j=mid+1;
while(i<=mid && j<=r)
{
if((long)help[i-l]>2*(long)help[j-l])
{
res+=mid-i+1; //翻转对个数
j++;
}
else i++;
}
//下面进行归并排序
i=l;
j=mid+1;
for(int k=l;k<=r;k++)
{
if(i>mid)
{
nums[k]=help[j-l];
j++;
}
else if(j>r)
{
nums[k]=help[i-l];
i++;
}
else if(help[i-l]>help[j-l])
{
nums[k]=help[j-l];
j++;
}
else
{
nums[k]=help[i-l];
i++;
}
}
return res;
}
int _reversepairs(vector<int>& nums,int l,int r)
{
if(l>=r)
return 0;
int mid=l+(r-l)/2;
int res1=_reversepairs(nums,l,mid);
int res2=_reversepairs(nums,mid+1,r);
int res3=merge(nums,l,mid,r);
return res1+res2+res3;
}
int reversePairs(vector<int>& nums) {
int n=nums.size();
return _reversepairs(nums,0,n-1);
}
};
注意:
统计翻转对的数量时,不能向剑指offer51题一样和归并排序一趟做了,因为翻转对不是简单的大小关系的比较。
统计翻转对的时机:得到左右两个有序数组之后,将两个数组归并成一个数组之前。