面试题51. 数组中的逆序对

题目描述

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/shu-zu-zhong-de-ni-xu-dui-lcof

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

示例 1:

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

思路

可以暴力，但是这道题的题解是用归并排序

代码

class Solution {
public:
    int mergeSort(vector<int>& nums, vector<int>& temp, int l, int r){
        if(l >= r)
            return 0;
        
        int mid = (l + r) / 2;
        //将区间分为两个区间，到最后分成每一个只有一个元素的区间
        int count = mergeSort(nums,temp,l,mid) + mergeSort(nums,temp,mid + 1,r);
        int i = l;
        int j = mid + 1;
        //将子区间进行排序向上合并，同时算出这一层的逆序对数。
        for(int k = l; k <= r; k++){
            if(i <= mid && (j > r || nums[i] <= nums[j])) {
                temp[k] = nums[i++];
                count += (j - mid -1);
            }else{
                temp[k] = nums[j++];
            }
        }
        for(int m = l; m <= r; m++)
            nums[m] = temp[m];
        return count;      
    }
    int reversePairs(vector<int>& nums) {
        int n = nums.size();
        vector<int> temp(n);
        return mergeSort(nums,temp,0,n-1);
    }
};