Leetcode---数组中的逆序对---树状数组 + 离散化

面试题51. 数组中的逆序对

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

示例 1:

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

限制：

0 <= 数组长度 <= 50000

思路

离散化 + 反向构建树状数组

实现代码

class BIT{
private:
    vector<int> tree;
    int len;
public:
    BIT(int n):len(n), tree(n+1,0){}
    
    static int lowbit(int x){
        return x & -x;
    }
    
    int query(int x){
        int res = 0;
        for(int i = x; i > 0; i -= lowbit(i)){
            res += tree[i];
        }
        return res;
    }
    
    void update(int x){
        for(int i = x; i <= len; i += lowbit(i)){
            tree[i]++;
        }
    }
    
};


class Solution {
public:
    int reversePairs(vector<int>& nums) {
        int len = nums.size(), ans = 0;
        if(!len) return ans;
        vector<int> tmp = nums;
        sort(tmp.begin(), tmp.end());
        for(int &num : nums){
            num = lower_bound(tmp.begin(), tmp.end(), num) - tmp.begin() + 1; // 离散化
        }
        BIT bit(len);
        for(int i = len - 1; i >= 0; i--){
            ans += bit.query(nums[i] - 1);
            bit.update(nums[i]);
        }
        return ans;
    }
};