LeetCode 315. 计算右侧小于当前元素的个数 & 剑指 Offer 51. 数组中的逆序对

题目

给定一个整数数组 nums，按要求返回一个新数组 counts。数组 counts 有该性质： counts[i] 的值是 nums[i] 右侧小于 nums[i] 的元素的数量。

示例:

输入:
[5,2,6,1]
输出:
[2,1,1,0]
解释:
5 的右侧有 2 个更小的元素 (2 和 1).
2 的右侧仅有 1个更小的元素 (1).
6 的右侧有 1 个更小的元素 (1).
1 的右侧有 0 个更小的元素.

解题思路

主要思路：运用归并排序的过程中，合并两个有序数组的操作可以计算逆序对。

• 即在合并第一个数组的时候记录：它右部的有序数组中有几个数已经被合并过了。
• 这个题目需要我们找出每个元素的逆序对，故需要在排序过程中标识当前的元素是数组中第几个元素以便求出逆序对。

代码

class Solution {
public:
    vector<int> countSmaller(vector<int>& nums) {
        //归并排序，排序过程中，求出逆序的个数
        int n = nums.size();
        if(n == 0){
            return {};
        }
        //需要用到辅助数组indexs来记录下标，以便于求出nums[i]的逆序数
        vector<int> temp(n);
        vector<int> indexs(n);
        //用于返回的结果
        vector<int> counts(n);   
        for(int i = 0; i < n; ++i){
            indexs[i] = i;
            counts[i] = 0;
        }
        reversePairs(nums, indexs, 0, n - 1, temp, counts);
        return counts;
    } 

    void reversePairs(vector<int> &nums, vector<int> &indexs, int left, int right, vector<int> &temp, vector<int> &counts){
        if(left == right){
            return;
        }
        int mid = left + (right - left) / 2;
        reversePairs(nums, indexs, left, mid, temp, counts);
        reversePairs(nums, indexs, mid + 1, right, temp, counts);

        if(nums[indexs[mid]] <= nums[indexs[mid + 1]]){
            return;
        }

        merge(nums, indexs, left, mid, right, temp, counts);
    }

    void merge(vector<int> &nums, vector<int> &indexs, int left, int mid, int right, vector<int> &temp, vector<int> &counts){
        //先将数据拷贝到temp
        for(int i = left; i <= right; ++i){
            temp[i] = indexs[i];
        }

        int i = left;
        int j = mid + 1;
        //注：在前面执行归并的时候计算逆序对
        for(int k = left; k <= right; ++k){
            if(i == mid + 1){
                indexs[k] = temp[j];
                ++j;
            }
            else if(j == right + 1){
                indexs[k] = temp[i];
                ++i;
                counts[indexs[k]] += j - mid - 1;
            }
            else if(nums[temp[i]] <= nums[temp[j]]){
                indexs[k] = temp[i];
                ++i;
                counts[indexs[k]] += j - mid - 1;
            }
            else{
                indexs[k] = temp[j];
                ++j;
            }
        }
    }
};