树状数组的区间修改、区间查询

最新推荐文章于 2024-09-18 20:54:10 发布

wennitao

最新推荐文章于 2024-09-18 20:54:10 发布

阅读量798

点赞数 2

分类专栏：算法文章标签：树状数组数据结构算法

本文链接：https://blog.csdn.net/wennitao222/article/details/90145871

版权

算法专栏收录该内容

28 篇文章 0 订阅

订阅专栏

设原数组第 $i$ 位为 $a [i]$ ， $a [0] = 0$ ， $d [i] = a [i] - a [i - 1]$

那么 $a[x]=\sum _{i=1} ^{x} d[i]$

$\sum _{i=1} ^{x} a[i] = \sum _{i=1} ^{x} \sum _{j=1} ^{i} d[j] = \sum _{i=1} ^{x} (x-i+1)d[i] = (x+1) \sum _{i=1} ^{x}d[i] - \sum _{i=1} ^{x} i * d[i]$

即原数组差分后，维护 $\sum _{i=1} ^{x}d[i]$ 和 $\sum _{i=1} ^{x} i * d[i]$ 即可

inline int lowbit (int x) {return x & (-x);}
inline void update (int pos, ll val) {
    for (int i = pos; i <= n; i += lowbit (i)) {
        c1[i] += val; c2[i] += 1ll * pos * val ;
    }
}
inline ll cal (int pos) {
    ll res = 0 ;
    for (int i = pos; i; i -= lowbit (i))
        res += c1[i] * (pos + 1) - c2[i] ;
    return res ;
}