leetcode 面试题51. 数组中的逆序对 官答笔记

题目链接

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

示例 1:

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

代码：

class Solution {
public:
    int mergeSort(vector<int>& nums, vector<int>& tmp, int l, int r) {
        if (l >= r) {
            return 0;
        }

        int mid = (l + r) / 2;
        int inv_count = mergeSort(nums, tmp, l, mid) + mergeSort(nums, tmp, mid + 1, r);
        int i = l, j = mid + 1, pos = l;
        while (i <= mid && j <= r) {
            if (nums[i] > nums[j]) {
                tmp[pos] = nums[j];
                ++j;
                inv_count += (mid-i+1);
            }
            else {
                tmp[pos] = nums[i];
                ++i;
            }
            ++pos;
        }
        while (i <= mid) {
            tmp[pos++] = nums[i];
            ++i;
        }
        while (j <= r) {
            tmp[pos++] = nums[j];
            ++j;
        }
        copy(tmp.begin() + l, tmp.begin() + r + 1, nums.begin() + l);
        return inv_count;
    }

    int reversePairs(vector<int>& nums) {
        int n = nums.size();
        vector<int> tmp(n);
        return mergeSort(nums, tmp, 0, n - 1);
    }
};

分析：
分治思想，归并排序
示例数组：[2,3,5,7,1,4,6,8]
该示例的前四个元素和后四个元素分别构成递增序列。
如果要计算该数组的逆序对，可以使用归并排序的方式，将[2,3,5,7]和[1,4,6,8]（视为数组1和数组2）并成一个数组（搞俩指针p1，p2）。当p2<p1时，此时p2的元素要移到新数组最前头，跨越了几个元素？跨越了数组1内的剩余元素个数（此处是4个），所以p2对应元素移动时可以看到去掉了4个逆序对，计数变量count此时加个4。
数组1和数组2变成了[2,3,5,7]和[4,6,8]，此时p2>p1，那2就放在新数组里，但此时它并没有跨越任何元素，因此不加，就这么一直算下去。

示例数组：[8,7,6,5,4,3,2,1]
这里就是归并排序的递归调用了，两两相比形成4个size为2的数组[7,8],[5,6],[3,4],[1,2]。此时count总共加了4。接着继续按上述方式调用，最终就得到了答案。

接下来考虑特殊情况：
（1）数组1元素先插完了，数组2剩一堆，这个没影响，继续插到新数组里就行，为什么呢？因为count计算的是跨越了数组1内的剩余元素个数，数组1空了那你就没得跨了，就没有去掉本存在于原数组里的逆序对。
（2）数组2元素先插完了，数组1本就在开头，插进去时有逆序对也是在数组2元素插入时就计算过了，因此它不用count计数了。