剑指offer——数组中的逆序对（c++)

题目描述

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

输入：[1,2,3,4,5,6,0]
输出：6

思路

先把数组分隔成子数组，统计出子数组内部的逆序对的数目，然后在统计出两个相邻子数组之间的逆序对的数目。合并完两个已排序的子数组后，要更新数组。
我们以数组{7,5,6,4}来分析统计逆序对的过程。先把数组分解成两个长度为2的子数组， 再把这两个子数组分别拆分成两个长度为1 的子数组。接下来一边合并相邻的子数组， 一边统计逆序对的数目。在第一对子数组中（7, 5）组成一个逆序对。同样在第二对子数组中也有逆序对（6, 4）。由于我们已经统计了这两对子数组内部的逆序对，因此需要把这两对子数组排序（ 图5.2 ( c）所示），以免在以后的统计过程中再重复统计。

接下来统计两个长度为2的子数组之间的逆序对，先用两个指针分别指向两个子数组的末尾，并每次比较两个指针指向的数字。如果第一个子数组中的数字大于第二个子数组中的数字，则构成逆序对，并且逆序对的数目等于第二个子数组中剩余数字的个数（如图5.2 (a)和图5.2 ©所示）。如果第一个数组中的数字小于或等于第二个数组中的数字，则不构成逆序对（如图5.3 (b)所示〉。每一次比较的时候，我们都把较大的数字从·后往前复制到一个辅助数组中去，确保辅助数组中的数字是递增排序的。在把较大的数字复制到辅助数组之后，把对应的指针向前移动一位，接下来进行下一轮比较。

注：图中省略了最后一步， 即复制第二个子数组最后剩余的4 到辅助数组中。
(a) $P_1$指向的数字大于$P_2$指向的数字，表明数组中存在逆序对。$P_2$ 指向的数字是第二个子数组的第二个数字， 因此第二个子数组中有两个数字比7 小。把逆序对数目加2，并把7 复制到辅助数组，向前移动$P_1$和$P_3$。
(b) $P_1$指向的数字小子$P_2$ 指向的数字，没有逆序对。把$P_2$ 指向的数字复制到辅助数组，并向前移动$P_2$ 和$P_3$ 。
© $P_1$指向的数字大于$P_2$ 指向的数字，因此存在逆序对。由于$P_2$ 指向的数字是第二个子数组的第一个数字，子数组中只有一个数字比5 小。把逆序对数目加1 ，并把5复制到辅助数组，向前移动$P_1$和$P_3$ 。

代码

class Solution {
public:
    int inversePairs(vector<int>& nums) {
        int n = nums.size();
        if(n < 2)
            return 0;
        else{
            return getCount(nums, 0, n-1);
        }
    }
    int getCount(vector<int>& nums, int begin, int end){
        if(begin == end)
            return 0;
        int mid = (begin + end) / 2;
        // copy: 辅助数组
        vector<int> copy(nums); 
        // 递归，归并排序，并计算本次逆序对数
        int leftCount = getCount(nums, begin, mid);
        int rightCount = getCount(nums, mid+1, end);
        // 记录交叉的逆序对数
        int crossCount = 0;
        //i：前半部分的下标，j：后半部分的下标，tempIdx：辅助数组的下标
        int i = mid, j = end, index = end;
        while(i >= begin && j > mid){
        // 存在交叉的逆序对，先统计一下，然后依次将较大值放进辅助数组
            if(nums[i] > nums[j]){
                copy[index--] = nums[i--];
                crossCount += j - begin - (mid - begin);
            }
            // 不存在交叉的逆序对，依次将较大值放进辅助数组
            else{
                copy[index--] = nums[j--];
            }
        }
        while(i >= begin){
            copy[index--] = nums[i--];
        }
        while(j > mid){
            copy[index--] = nums[j--];
        }
        //更新归并排序后的子数组
        for(int i = begin; i <= end; ++i)
            nums[i] = copy[i];
        return (leftCount+rightCount+crossCount);
    }
};