力扣493.翻转对

方法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题一样和归并排序一趟做了,因为翻转对不是简单的大小关系的比较。

统计翻转对的时机:得到左右两个有序数组之后,将两个数组归并成一个数组之前。

猜你喜欢

转载自blog.csdn.net/m0_38062470/article/details/113881112