一维树状数组

hide_on-BUSh

于 2023-03-28 21:07:31 发布

阅读量69

点赞数

文章标签：算法数据结构 java

本文链接：https://blog.csdn.net/m0_62909934/article/details/129825098

版权

文章介绍了树状数组这一高级数据结构，主要用于处理单点修改和区间查询的问题。通过区间修改和单点查询的函数实现，展示了树状数组在处理区间求和时的时间和空间效率优势，对比了其与带lazy标记的线段树的优劣。

摘要由CSDN通过智能技术生成

“高级”数据结构——树状数组！

1. 单点修改 + 区间查询

最简单的树状数组就是这样的：

void add(int p, int x){ //给位置p增加x
    while(p <= n) sum[p] += x, p += p & -p;
}
int ask(int p){ //求位置p的前缀和
    int res = 0;
    while(p) res += sum[p], p -= p & -p;
    return res;
}
int range_ask(int l, int r){ //区间求和
    return ask(r) - ask(l - 1);
}

2. 区间修改 + 单点查询

通过“差分”（就是记录数组中每个元素与前一个元素的差），可以把这个问题转化为问题1。

void add(int p, int x){ //这个函数用来在树状数组中直接修改
    while(p <= n) sum[p] += x, p += p & -p;
}
void range_add(int l, int r, int x){ //给区间[l, r]加上x
    add(l, x), add(r + 1, -x);
}
int ask(int p){ //单点查询
    int res = 0;
    while(p) res += sum[p], p -= p & -p;
    return res;
}

3. 区间修改 + 区间查询

void add(ll p, ll x){
    for(int i = p; i <= n; i += i & -i)
        sum1[i] += x, sum2[i] += x * p;
}
void range_add(ll l, ll r, ll x){
    add(l, x), add(r + 1, -x);
}
ll ask(ll p){
    ll res = 0;
    for(int i = p; i; i -= i & -i)
        res += (p + 1) * sum1[i] - sum2[i];
    return res;
}
ll range_ask(ll l, ll r){
    return ask(r) - ask(l - 1);
}

用这个做区间修改区间求和的题，无论是时间上还是空间上都比带lazy标记的线段树要优。