LeetCode 剑指 Offer 51. 数组中的逆序对--归并排序分治优化

剑指 Offer 51. 数组中的逆序对
在数组中的两个数字，如果前面一个数字大于后面的数字，则这两个数字组成一个逆序对。输入一个数组，求出这个数组中的逆序对的总数。

示例 1:

输入: [7,5,6,4]
输出: 5

限制：

0 <= 数组长度 <= 50000

题解

利用归并排序的递归处理过程记录答案。

AC代码

class Solution {
public:
    int ans;
    void Sort(vector<int>&nums,int l,int r)
    {
        if(l>=r)return;
        int mid=(l+r)/2;
        Sort(nums,l,mid);
        Sort(nums,mid+1,r);
        vector<int>q1,q2;
        for(int i=l;i<=mid;i++)
        q1.push_back(nums[i]);
        for(int i=mid+1;i<=r;i++)
        q2.push_back(nums[i]);
        int s1=0,s2=0,sign=l;
        while(s1<q1.size()&&s2<q2.size())
        {
            if(q1[s1]>q2[s2])
            {
                nums[sign]=q1[s1];
                s1++;
            }
            else
            {
                nums[sign]=q2[s2];
                s2++;
                ans+=s1;//前面的s1个都是大于q2[s2]的，所以可以构成s1个逆序对
            }
            sign++;
        }
        while(s1<q1.size())
        {
            nums[sign]=q1[s1];
            s1++;
            sign++;
        }
        while(s2<q2.size())
        {
            nums[sign]=q2[s2];
            s2++;
            sign++;
            ans+=s1;//前面的s1个都是大于q2[s2]的，所以可以构成s1个逆序对
        }
    }
    int reversePairs(vector<int>& nums) {
        if(nums.size()<2)return 0;
        ans=0;
        Sort(nums,0,nums.size()-1);
        return ans;
    }
};