Leetcode---315. 计算右侧小于当前元素的个数---每日一题---树状数组+离散化

315. 计算右侧小于当前元素的个数

给定一个整数数组 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 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:
    vector<int> countSmaller(vector<int>& nums) {
        int len = nums.size();
        BIT bit(len);
        vector<int> res(len, 0);
        // 离散化
        vector<int> tmp = nums;
        sort(tmp.begin(), tmp.end());
        for (int& num : nums) {
            num = lower_bound(tmp.begin(), tmp.end(), num) - tmp.begin() + 1;
        }

        // 使用树状数组
        for (int i = len - 1; i >= 0; i--) {
            res[i] = bit.query(nums[i] - 1);
            bit.update(nums[i]);
        }
        return res;
    }
};