leetcode树状数组

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 Solution {
private:
    // 树状数组
    vector<int> c;
    void Init(int length){
        c.resize(length,0);
    }
    int LowBit(int x){
        return x&(-x);
    }
    void Updata(int pos){
        while(pos<c.size()){
            c[pos]+=1;
            pos+=LowBit(pos);
        }
    }
    int Query(int pos){
        int ret=0;

        while(pos>0){
            ret+=c[pos];
            pos-=LowBit(pos);
        }
        return ret;
    }

    // 离散化
    vector<int> a;
    void Discretization(vector<int>& nums){
        a.assign(nums.begin(),nums.end());
        sort(a.begin(),a.end());
        a.erase(unique(a.begin(),a.end()),a.end());
    }
    int getId(int x){
        return lower_bound(a.begin(),a.end(),x)-a.begin()+1;
    }

public:
    vector<int> countSmaller(vector<int>& nums) {
        vector<int> resultList;

        Discretization(nums);
        Init(nums.size()+5);

        for(int i=(int)nums.size()-1;i>=0;i--){
            int id=getId(nums[i]);
            resultList.push_back(Query(id-1));
            Updata(id);
        }
        reverse(resultList.begin(),resultList.end());

        return resultList;
    }
};