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;
}
};