LeetCode315 计算右侧小于当前元素的个数（分治求逆序数）

题目链接：leetcode315

给你一个数组，让你求每个位置的数后面比它小的数的数量。也就是以每个数字为头的逆序对的数量。

解题思路

树状数组、线段树

先离散化一下，然后逆序统计每个位置前比该数字小的数。

分治

先离散化，再利用归并排序把数组逆序，并求后面的区间对前面区间的贡献。

代码

分治

class Solution {
public:
    vector<int> countSmaller(vector<int>& nums) {
        int n = nums.size();
        vector<int> countArray(n, 0);
        vector<pair<int, int> > arr;
        for (int i = 0; i < n; i++) arr.push_back({nums[i], i}); //离散化
        MergeSort(0, n-1, arr, countArray);
        return countArray;
    }
    
    void MergeSort(int l, int r, vector<pair<int, int> >& arr, vector<int>& countArray) {
        if (l >= r) return ;
        
        int mid = l + (r - l >> 1);
        MergeSort(l, mid, arr, countArray);        
        MergeSort(mid + 1, r, arr, countArray);
        int i = l, j = mid + 1;
        vector<pair<int, int> > tmp;
        while (i <= mid && j <= r) {
            if (arr[i].first > arr[j].first) {
                tmp.push_back(arr[i]);
                countArray[arr[i].second] += r - j + 1;
                i++;
            } else {
                tmp.push_back(arr[j]);
                j++;                
            }
        }
        
        while (i <= mid) tmp.push_back(arr[i++]);
        while (j <= r) tmp.push_back(arr[j++]);
        
        for (i = l; i <= r; i++) arr[i] = tmp[i-l];
    }
};