树状数组

一、前缀和与差分

1. 原数组：$a_1,a_2,a_3,....,a_n$
2. 前缀和：$S_i={\sum }_{k=1}^{k=i}{a}_i$，$a_i=S_i-S_{i-1}$
3. 差分数组：$X_i=a_i-a_{i-1}$
4. X 数组是 a 数组的差分数组，a 数组是 S 数组的差分数组
5. S 数组是 a 数组的前缀和数组，a 数组是 X 数组的前缀和数组
6. 前缀和数组以及差分数组，并没有增加信息，只是信息的另外一种表示形式
7. 前缀和数组用来优化区间和操作
8. 差分数组用来优化区间修改操作

问题1：原数组区间和操作

a 数组上的操作：$O(n)$

S 数组上的操作：$O(1)，S_i-S_{j-1}=a[j,i]区间和$

问题2：原数组区间元素修改（加法）

a = { a 1 , a 2 , a 3 , a 4 , a 5 , a 6 } a=\{a_1,a_2,a_3,a_4,a_5,a_6\} a={a1​,a2​,a3​,a4​,a5​,a6​}

X = { X 1 = a 1 − a 0 , X 2 = a 2 − a 1 , X 3 = a 3 − a 2 , X 4 = a 4 − a 3 , X 5 = a 5 − a 4 , X 6 = a 6 − a 5 } X=\{X_1=a_1-a_0,X_2=a_2-a_1,X_3=a_3-a_2,X_4=a_4-a_3,X_5=a_5-a_4,X_6=a_6-a_5\} X={X1​=a1​−a0​,X2​=a2​−a1​,X3​=a3​−a2​,X4​=a4​−a3​,X5​=a5​−a4​,X6​=a6​−a5​}

{ a 1 , a 2 + d , a 3 + d , a 4 + d , a 5 , a 6 } \{a_1,a_2+d,a_3+d,a_4+d,a_5,a_6\} {a1​,a2​+d,a3​+d,a4​+d,a5​,a6​}

X = { X 1 = a 1 − a 0 , X 2 = a 2 − a 1 + d , X 3 = a 3 − a 2 , X 4 = a 4 − a 3 , X 5 = a 5 − a 4 − d , X 6 = a 6 − a 5 } X=\{X_1=a_1-a_0,X_2=a_2-a_1+d,X_3=a_3-a_2,X_4=a_4-a_3,X_5=a_5-a_4-d,X_6=a_6-a_5\} X={X1​=a1​−a0​,X2​=a2​−a1​+d,X3​=a3​−a2​,X4​=a4​−a3​,X5​=a5​−a4​−d,X6​=a6​−a5​}

a 数组时间复杂度：$O(n)$

X 数组时间复杂度：$O(1)$

二、树状数组

1. lowbit 函数求数字 i，二进制表示中的最低1所在的位权
2. lowbit(x) = x & (-x)
3. 树状数组本质上是对前缀和数组的一种优化，主要体现在单点修改操作上
4. 前缀和查询 $O(logn)$，单点修改$O(logn)$
5. 相比于最普通的前缀和数组，查询方面变差，单点修改操作变好，综合时间复杂度变好
6. 查询的时候，向前统计，$i$ 的前一位 $i-lowbit(i)$
7. 修改的时候，向后修改，$i$ 的后一位 $i+lowbit(i)$
   
   
   
   

三、海贼 OJ-329-弱化的整数问题

题目描述

给定长度为N(N≤100000)的序列A(0≤Ai≤100000)，然后输入m(m≤105)行操作指令

第一类指令形如C l r d(1≤l≤r≤N)，表示把数列中第l…r之间的数都加d(0≤d≤100000)

第二类指令形如Q x(x≤N)，表示询问序列中第x个数的值。

---

输入

第一行一个整数N,代表序列A的长度

第二行是由空格分隔开的N个数，分别代表A1，A2……An

接下来一行是一个整数m，代表操作的次数。

接下来m行，每行代表这一条指令如题目所述

输出

对于每次Q查询，输出一行为查询的值。

输入样例1

5
1 5 3 2 4
2
C 1 5 1
Q 3

输出样例1

4

---

数据规模与限定

时间限制：1 s

内存限制：64 M

引入差分数组

a = { a 1 , a 2 , a 3 , a 4 , a 5 , a 6 } a=\{a_1,a_2,a_3,a_4,a_5,a_6\} a={a1​,a2​,a3​,a4​,a5​,a6​}

X = { X 1 = a 1 − a 0 , X 2 = a 2 − a 1 , X 3 = a 3 − a 2 , X 4 = a 4 − a 3 , X 5 = a 5 − a 4 , X 6 = a 6 − a 5 } X=\{X_1=a_1-a_0,X_2=a_2-a_1,X_3=a_3-a_2,X_4=a_4-a_3,X_5=a_5-a_4,X_6=a_6-a_5\} X={X1​=a1​−a0​,X2​=a2​−a1​,X3​=a3​−a2​,X4​=a4​−a3​,X5​=a5​−a4​,X6​=a6​−a5​}

{ a 1 , a 2 + d , a 3 + d , a 4 + d , a 5 , a 6 } \{a_1,a_2+d,a_3+d,a_4+d,a_5,a_6\} {a1​,a2​+d,a3​+d,a4​+d,a5​,a6​}

X = { X 1 = a 1 − a 0 , X 2 = a 2 − a 1 + d , X 3 = a 3 − a 2 , X 4 = a 4 − a 3 , X 5 = a 5 − a 4 − d , X 6 = a 6 − a 5 } X=\{X_1=a_1-a_0,X_2=a_2-a_1+d,X_3=a_3-a_2,X_4=a_4-a_3,X_5=a_5-a_4-d,X_6=a_6-a_5\} X={X1​=a1​−a0​,X2​=a2​−a1​+d,X3​=a3​−a2​,X4​=a4​−a3​,X5​=a5​−a4​−d,X6​=a6​−a5​}

引入差分数组 X，将原数组 a 上的区间加操作，转换成 X 数组上的两次【单点操作】

对于查询原数组 a[i] 的值，等价于查询 X 数组前 i 位的【前缀和】

结论

由于，既要维护【前缀和】，又要进行【单点修改】，所以可以使用树状数组

代码实现

#include<iostream>
#include <cstdio>
using namespace std;
#define MAX_N 100000
int c[MAX_N + 5];
#define lowbit(x) (x & -x)

void add(int i, int x, int n)
{
    while (i <= n) {
        c[i] += x;
        i += lowbit(i);
    }
    return;
}

int query(int i) {
    int sum = 0;
    while (i) {
        sum += c[i];
        i -= lowbit(i);
    }
    return sum;
}

int main(int argc, char *grav[])
{
    int m, n, pre = 0, a;
    scanf("%d", &n);
    for (int i = 1;i <= n; i++) {
        scanf("%d", &a);
        add(i, a - pre, n);
        pre = a;
    }
    scanf("%d", &m);
    char s[10];
    int l, r, x;
    for (int i = 0; i < m; i++) {
        scanf("%s", s);;
        switch (s[0]) {
            case 'C': {
                scanf("%d%d%d", &l, &r, &x);
                add(l, x, n);
                add(r + 1, -x, n);
            } break;
            case 'Q': {
                scanf("%d", &x);
                printf("%d\n", query(x));
            } break;
        }
    }
    return 0;
}