前言
欢迎大家积极在评论区留言发表自己的看法,知无不言,言无不尽,养成每天刷题的习惯,也可以自己发布优质的解题报告,供社区一同鉴赏,吸引一波自己的核心粉丝。
今天是七月集训第二十五天:树状数组🔥🔥🔥🔥
一、练习题目
二、算法思路
- 1、327. 区间和的个数:🔥🔥🔥🔥我今天先学了树状数组的内容,这道题使用的树状数组,我的代码是直接参考大佬的代码码写的。
三、源码剖析
// 327. 区间和的个数
class Solution {
#define ll long long
ll s[100010];
vector<ll> bin;
int c[300010];
int getIndex(ll val) {
int l = 0, r = bin.size() - 1;
while(l <= r) {
int mid = (l + r) >> 1;
if(val > bin[mid]) {
l = mid + 1;
}else if(val < bin[mid]) {
r = mid - 1;
}else {
return mid + 1;
}
}
return -1;
}
int lowbit(int x) {
return x & -x;
}
void add(int x, int n) {
while(x <= n) {
c[x]++;
x += lowbit(x);
}
}
ll sum(int x) {
ll s = 0;
while(x) {
s += c[x];
x -= lowbit(x);
}
return s;
} //(1)
public:
int countRangeSum(vector<int>& nums, int l, int u) {
int n = nums.size();
int ans = 0;
bin.clear();
memset(c, 0, sizeof(c));
s[0] = 0;
for(int i = 1; i <= n; ++i) {
s[i] = s[i - 1] + nums[i - 1];
} //前缀和
for(int i = 0; i <= n; ++i) {
bin.push_back(s[i]);
bin.push_back(s[i] - u);
bin.push_back(s[i] - l);
}
sort(bin.begin(), bin.end());
bin.erase(unique(bin.begin(), bin.end()), bin.end());
add(getIndex(s[0]), bin.size());
for(int i = 1; i <= n; ++i) {
ans += sum(getIndex(s[i] - l)) - sum(getIndex(s[i] - u) - 1);
add(getIndex(s[i]), bin.size());
}
return ans;
}
};
- 1、这个可以说用到树状数组就是比较经典的模板了:
1.add:在树状数组的x位置增加u;
2、sum用于求和的操作;