LeetCode-493. 翻转对、归并排序

给定一个数组 nums ，如果 i < j 且 nums[i] > 2*nums[j] 我们就将 (i, j) 称作一个重要翻转对。
你需要返回给定数组中的重要翻转对的数量。
示例 1:
输入: [1,3,2,3,1]
输出: 2
来源：力扣（LeetCode）

题目分析

如果 i 、j 分别属于两个有序区间，并且 nums[i] > 2*nums[j] ，则大于 i 的元素也都满足需求，利用有序特点可以减少重复的比较操作。

归并排序

将两个有序的数组合并成一个有序数组称为归并。归并排序包含了两个过程：
①：从上往下的分解：把当前区间一分为二，直至分解为若干个长度为1的子数组
②：从下往上的合并：两个有序的子区域两两向上合并
归并排序的一般步骤：

①．分解：把当前区间一分为二，分解点即中间点mid = (left+right)/2
②．求解：分别递归左右两个子区间[left…mid] 和 [mid+1…right]进行归并排序。递归的终结条件是子区间长度为1。
③．合并：把两个有序子数组合并需要占用一个临时空间，依次挪动两个子区间的指针，比较元素值大小，将较小的值存入临时空间的开头。将两个有序区间归并成一个临时有序区间[left…right]，并将结果拷贝到原数组的区间[left…right]，使原始数组[left…right]变为有序。

代码示例

对于本题而言，在归并之前进行翻转对的数量统计即可。

class Solution {
public:
    int reversePairs(vector<int>& nums) {
        int size = nums.size();
        int left = 0;
        int right = size-1;
        int result = 0;
        vector<int> temp(size);
        mergesort(nums,temp,left,right,result);
        return result;
    }
    void mergesort( vector<int>& nums ,vector<int>& temp,int left,int right,int &result)
    {
        if( left >= right) return ;//基线条件，只剩一个元素就是有序的了
        int mid = left + (right-left)/2;//分隔点
        mergesort(nums,temp,left,mid,result);//对左区间排序
        mergesort(nums,temp,mid+1,right,result);//对右区间排序
        
        int i = left;
        int j = mid+1;
        while( i <= mid && j <= right )//左右区间都是有序的了，进行统计操作
        {
            if( (long long )nums[i] > (long long )2*nums[j] )// *2 会超 int 范围
            {
                result += ( mid-i +1);//反转对个数
                j++;
            }
            else
            {
                i++;
            }
        }

        i = left;
        j = mid+1;
        int t = 0;
        while( i <= mid && j <= right )//进行归并操作
        {
            if( nums[i] > nums[j] )
            {
                temp[t++] = nums[j++];
            }
            else
            {
                temp[t++] = nums[i++] ;
            }
        }
        while( i <= mid )//将左边剩余元素填充进temp中
        {
            temp[t++] = nums[i++];
        }
        while( j <= right )//将右序列剩余元素填充进temp中
        {
            temp[t++] = nums[j++];
        }
        t = 0;
        while( left <= right )//将temp中的元素全部拷贝到原数组[left,right]中
        {
            nums[left++] = temp[t++] ;
        }
    }
};