剑指offer-数组中的逆序对-JavaScript

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

解法 1: 暴力法(TLE)

直接双重循环,挨个检查是否为逆序对。代码实现比较简单:

/**
 * @param {number[]} nums
 * @return {number}
 */
var reversePairs = function(nums) {
    let res = 0;
    const length = nums.length;
    for (let i = 0; i < length; ++i) {
        for (let j = i + 1; j < length; ++j) {
            nums[i] > nums[j] && ++res;
        }
    }

    return res;
};

时间复杂度是。在 leetcode 上会 TLE,无法通过(毕竟这是道标注「困难」的题目)。

解法 2: 归并排序(正确解法)

这题的正确解法是要借助归并排序的思路,在归并的过程中,快速统计逆序对。这种解法比较难想到,但是应用归并排序的题目真的不多,所以这题很有研究和收藏意义。

核心的解决逻辑都封装在 findInversePairNum 函数中。它的职能就是统计数组arr[start, end]范围中的逆序对,并且统计完后,arr[start, end]范围中的元素会被排序(这点和归并排序的过程一样)。

那么函数又是如何快速统计逆序对的呢?大体过程如下:

  • 递归调用,拿到左子数组和右子数组的逆序对(此时,左子数组和右子数组也都排序完成了)
  • 指针 i 和 j 分别指向左子数组和右子数组的最右侧,此时会有 2 种情况:
    • arr[i] > arr[j]:那么说明arr[i]大于右子数组中所有元素,逆序对增加j - start - length,向左边移动指针 i
    • arr[i] <= arr[j]: 对arr[i]来说,不存在逆序对,向左边移动指针 j
  • i 和 j 遍历完各自数组后,最后返回逆序对之和即可

代码实现如下:

// ac地址:https://leetcode-cn.com/problems/shu-zu-zhong-de-ni-xu-dui-lcof/
// 原文地址:https://xxoo521.com/2020-03-12-reverse-pairs/
/**
 * @param {number[]} nums
 * @return {number}
 */
var reversePairs = function(nums) {
    return findInversePairNum(nums, 0, nums.length - 1);
};

/**
 * @param {number[]} arr
 * @param {number} start
 * @param {number} end
 */
function findInversePairNum(arr, start, end) {
    if (start >= end) return 0;

    const copy = new Array(end - start + 1);
    const length = Math.floor((end - start) / 2); // 左数组长度
    const leftNum = findInversePairNum(arr, start, start + length);
    const rightNum = findInversePairNum(arr, start + length + 1, end);

    let i = start + length;
    let j = end;
    let copyIndex = end - start;
    let num = 0;
    while (i >= start && j >= start + length + 1) {
        if (arr[i] > arr[j]) {
            num += j - start - length;
            copy[copyIndex--] = arr[i--];
        } else {
            copy[copyIndex--] = arr[j--];
        }
    }

    while (i >= start) {
        copy[copyIndex--] = arr[i--];
    }

    while (j >= start + length + 1) {
        copy[copyIndex--] = arr[j--];
    }

    for (let k = start; k <= end; ++k) {
        arr[k] = copy[k - start];
    }

    return num + leftNum + rightNum;
}

时间复杂度是,空间复杂度是。如果还是觉得不好理解,可以以数组 7、5、6、4 为例,按照前面过程,手动计算一下。

更多资料

整理不易,若对您有帮助,请给个「关注+点赞」,您的支持是我更新的动力

  • Blog:剑指 Offer 题解 + JS 代码
  • Github :https://github.com/dongyuanxin/blog
  • 公众号:心谭博客

你可能感兴趣的:(剑指offer-数组中的逆序对-JavaScript)