剑指 51. 数组中的逆序对 - 难度困难

1. 题目描述

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

示例 1:

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

限制：

0 <= 数组长度 <= 50000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/shu-zu-zhong-de-ni-xu-dui-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 题解

1、归并排序、分治思想

思路：归并排序的 “分” 的阶段直到将数组拆分为单个元素后开始合并，因为我们认为单个元素是有序的；在“合并”阶段，待合并的左右子数组一定是部分有序的，如 [5,7] 和 [4,6]，基于这种部分有序，我们会发现逆序对的计数规律，即当 右子数组的元素 nums[j] 小于 左子数组的元素 nums[i] 时说明此时逆序对的个数是左子数组左边的元素个数 mid-i+1。

/**
 * @param {number[]} nums
 * @return {number}
 */
var reversePairs = function(nums) {
    const len = nums.length;
    if(len < 2){
        return 0;
    }
    const temp = new Array(nums.length);
    return reversePairsCore(nums, 0, nums.length-1, temp);
};

function reversePairsCore(nums, left, right, temp){
    if(left === right){
        return 0;
    }
    let mid = Math.floor(left + (right - left)/2);
    let leftPairs = reversePairsCore(nums, left, mid, temp);
    let rightPairs = reversePairsCore(nums, mid+1, right, temp);
    if(nums[mid]<= nums[mid+1]){
        return leftPairs + rightPairs;
    }
    let crossPairs = mergeAndCount(nums, left, mid, right, temp);
    return leftPairs + rightPairs + crossPairs;
}

function mergeAndCount(nums, left, mid, right, temp){
    let i = left;
    let j = mid+1;
    let t = 0;
    let count = 0; //逆序对的计数器
    while(i <= mid && j <= right){
        if(nums[i] <= nums[j]){
            temp[t++] = nums[i++]; 
        }else{
            temp[t++] = nums[j++];
            count += mid - i + 1; //计数
        }
    }
    //左半部分剩余元素填充到temp中
    while(i <= mid){
        temp[t++] = nums[i++];
    }
    //右半部分剩余元素填充到temp中
    while(j <= right){
        temp[t++] = nums[j++];
    }
	//根据temp修改nums数组
    t = 0;
    while(left <= right){
        nums[left++] = temp[t++];
    }
    return count;
}

时间复杂度：O(n*logn)；
空间复杂度：O(n)，开辟的辅助空间。