基本操作

树状数组 主要应用于单点修改和区间查询
核心操作就3个 lowbit \ add \ query

  1. lowbit
    lowbit 返回二进制最后一位的数

    1
    2
    3
    func lowbit(x)int{
        return x & -x
    }

  2. add
    为某一个位置x 加上一个数v (如果是减的话可以减去这个数的负数)

    1
    2
    3
    4
    5
    6
    7

    func add(x,v int){
        for i:= x ; i <= n ; i+=lowbit(x){
            tr[i] += v
        }
    }

  3. query
    查询 1-x区间的数组和

    1
    2
    3
    4
    5
    6
    7
    func query(x int)int{
        res := 0
        for i:= x ; i > 0 ; i-=lowbit(x){
            res += tri[i]
        }
        return res
    }

树状数组例题整理

只要符合两个条件 就可以重点考虑使用树状数组 单点修改和区间查询
当然符合其中一个条件也可以 只是没那个必要 均有更好的替代方法
难度1: 模板题 (很久没写找找感觉用)
动态求连续区间和

原题链接:
https://www.acwing.com/problem/content/1266/

acwing上的题用go 会有些问题 尤其是那些输入输出复杂的最好还是用c++ 当然也只可能时创始人比较懒没有做好适配…
比如这题… 用go就会有换行问题 最后居然也会超时就比较离谱

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
#include<iostream>
using namespace std;

const int N = 1E5 + 10;

int q[N],tr[N];
int n,m;

int lowbit(int x){
    return x & -x;
}

void add(int x,int v){
    for(int i=x ; i<=n; i+=lowbit(i)){
        tr[i] += v;       
    }
}

int query(int x){
    int res = 0;
    for(int i=x ; i > 0 ; i -= lowbit(i)){
        res += tr[i];
    }
    return res;
}

int main(){
    scanf("%d%d",&n,&m);
    for (int i = 1 ; i <= n ; i++){
        scanf("%d",&q[i]);
        add(i,q[i]);
    }
    
    for (int i=0 ; i < m ; i++){
        int k,a,b;
        scanf("%d%d%d",&k,&a,&b);
        if(!k){
            printf("%d\n",query(b) - query(a - 1));
        }else{
            add(a,b);
        }
    }

    return 0;
}