您的位置:首页 > 其它

(模板)树状数组 (区间修改,单点查询)

2018-03-13 18:07 387 查看
引入一个叫做查分数组的东西,感觉思想有点像lazytab。区间[l,r]所有值+k改成”位置l加上k,位置r+1减去k”

查询的时候直接查询sum(x)就行;(参考题目:洛谷P3386)

#include<cstdio>
#include<iostream>
using namespace std;
typedef long long ll;
const int maxn = 500050;
int a[maxn], c[maxn],n;//a是原始数组,c是差分数组
int sum(int i)//这两个函数与原始版的树状数组是一样的,只不过里面操作的是差分数组
{
int s = 0;
while (i) {
s += c[i];
i -= i&-i;
}
return s;
}
void add(int i, int x)
{
while (i <= n) {
c[i] += x;
i += i&-i;
}
}
int main(void)
{
int m, i, j;

cin >> n >> m;
for (i = 1; i <= n; i++)
scanf("%d", &a[i]);
for (i = 1; i <= m; i++) {
int e, f, c, d;
scanf("%d", &e);
if (e == 1) {
scanf("%d %d %d", &f, &c, &d);
add(f, d);
add(c + 1, -d);
}
else {
scanf("%d", &f);
printf("%d\n", a[f] + sum(f));//记得加上原来的数组
}
}

return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐