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

引入一个叫做查分数组的东西,感觉思想有点像lazytab。区间[l,r]所有值+k改成”位置l加上k,位置r+1减去k”
查询的时候直接查询sum(x)就行;(参考题目:洛谷P3386)

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

你可能感兴趣的:(模板,数据结构,树状数组)