(模板)树状数组 (区间修改,单点查询)
程序员文章站
2024-03-03 18:15:22
...
引入一个叫做查分数组的东西,感觉思想有点像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;
}
上一篇: PHP 接入微信扫码支付总结(总结篇)