元素守恒计算方法_树状数组求逆序对

元素守恒计算方法_树状数组求逆序对给定一个整数数组 nums 按要求返回一个新数组 counts 数组 counts 有该性质 counts i 的值是 nums i 右侧小于 nums i 的素的数量 示例 输入 nums 5 2 6 1 输出 2 1 1 0 解释 5 的右侧有 2 个更小的素 2 和 1 2 的右侧仅有 1 个更小的素 1 6 的右侧有 1 个更小的素

给定一个整数数组 nums,按要求返回一个新数组 counts。数组 counts 有该性质: counts[i] 的值是 nums[i] 右侧小于 nums[i] 的元素的数量。

示例:

输入:nums = [5,2,6,1]
输出:[2,1,1,0]
解释:
5 的右侧有 2 个更小的元素 (2 和 1)
2 的右侧仅有 1 个更小的元素 (1)
6 的右侧有 1 个更小的元素 (1)
1 的右侧有 0 个更小的元素

提示:

0 <= nums.length <= 10^5
-10^4 <= nums[i] <= 10^4

题解
树状数组求逆序数

class Solution { 

public:
static const int N = 1e5 + 10;
int lowbit(int x){

return (x & (-x));
}
int query(int x){

int sum = 0;
while(x > 0){

sum += tree[x];
x -= lowbit(x);
}
return sum;
}
void add(int x,int y){

while(x < N){

tree[x] += y;
x += lowbit(x);
}
}
int tree[N] = {
0};

vector countSmaller(vector& nums) {

vectorres;
int base = 1e4 + 1;
for(int i = nums.size() - 1;i >= 0;i --){

nums[i] = nums[i] + base;
res.push_back(query(nums[i] - 1));
add(nums[i],1);
}
reverse(res.begin(),res.end());
return res;
}
};
编程小号
上一篇 2025-10-09 11:46
下一篇 2025-04-24 16:27

相关推荐

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。
如需转载请保留出处:https://bianchenghao.cn/hz/144177.html