剑指offer51 数组中的逆序对

106 阅读1分钟

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

示例 1:

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

来源:力扣(LeetCode)
链接:leetcode-cn.com/problems/sh…
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

//在归并排序的基础上小小的改一下
class Solution {
public:
    vector<int>temp;
    int get_count(vector<int>&nums,int l,int r)
    {
        if(l>=r) return 0;
        int mid=(l+r)>>1;
        int ans=0;
        ans+=get_count(nums,l,mid);
        ans+=get_count(nums,mid+1,r);
        int left_pointer=l; int right_pointer=mid+1; int k=l;
        while(left_pointer<=mid||right_pointer<=r)
        {
            if((right_pointer>r)||left_pointer<=mid && nums[left_pointer]<=nums[right_pointer])
            temp[k++]=nums[left_pointer++];
            else {
                temp[k++]=nums[right_pointer++];
                ans+=(mid-left_pointer+1);
            }
        }
        for(int i=l;i<=r;i++) nums[i]=temp[i];
        return ans;
    }
    int reversePairs(vector<int>& nums) {
        while(temp.size()<nums.size()) temp.push_back(0);
        return get_count(nums,0,nums.size()-1);
    }
};