动态求连续区间和（树状数组 + 线段树详解）

题目描述：

给定 n 个数组成的一个数列，规定有两种操作，一是修改某个元素，二是求子数列 [a,b] 的连续和。

输入格式：

第一行包含两个整数 n 和 m，分别表示数的个数和操作次数。

第二行包含 n 个整数，表示完整数列。

接下来 m 行，每行包含三个整数 k,a,b（k=0，表示求子数列[a,b]的和；k=1，表示第 a个数加 b）。

数列从 1 开始计数。

输出格式：

输出若干行数字，表示 k=0 时，对应的子数列 [a,b] 的连续和。

数据范围：

1≤n≤100000
1≤m≤100000
1≤a≤b≤n

输入样例：

10 5
1 2 3 4 5 6 7 8 9 10
1 1 5
0 1 3
0 4 8
1 7 5
0 4 8

输出样例：

11
30
35

方法一    树状数组：

 树状数组主要应用于同时需要给某个位置上的数加上一个数（单点修改）、求某一个前缀和（区间查询）两个操作，两个操作的时间复杂度均为O（logn)。

树状数组还有许多变形的应用，如可以利用单点修改将某个位置的数变为某个数（即先加上这个数x的相反数-x，再加上需要得到的数v），也可以实现区间查询和单点修改，但较为复杂，此处不介绍。

对于同时需要单点修改和区间查询的题目来讲，使用树状数组效率更高。但如果题目只有区间和，直接使用前缀和更优。前缀和的缺点在于，如果要修改某一个数的值，最坏的情况下可能达到o(n)。

关于lowbit(x):    lowbit(x)是x的二进制表达式中最低位的1所对应的值。比如，6的二进制是110，所以lowbit(6)=2。

树状数组原理如下：

 树状数组的三个核心操作：

//  1. lowbit(x): 返回x的最后一位1
int lowbit(x){
    return x & -x;
}

//  2. add(int x, int v): 在x位置加上数v， 同时修改后面值中包含x的部分
void add(int x, int v){
    for(int i = x; i <= n; i += lowbit(i))  tr[i] += v;
}

//  3.query(int x): 询问x的前缀和
int query(int x){
    int res = 0;
    for(int i = x; i; i -= lowbit(i))   res += a[i];
    return res;
}

 树状数组的初始化问题：我们可以将初始化看为：在起始全为0的数组上的对应位置加上a[ i ]的操作，用add函数就可实现。

代码实现：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

const int N = 100010;

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

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; i -= lowbit(i))   res += tr[i];
    return res;
}

int main(){
    cin >> n >> m;
    for(int i = 1; i <= n; i ++ )   scanf("%d", &a[i]);
    for(int i = 1; i <= n; i ++ )   add(i, a[i]);

    while(m -- ){
        int k, x, y;
        scanf("%d%d%d", &k ,& x, &y);

        if(k == 0)  cout << query(y) - query(x - 1) << endl;
        else add(x, y);
    }

    return 0;
}

方法二    线段树：

线段树适用于适合树状数组的所有题目，除此之外，一些树状数组所不能解的题目也能完成。相较于树状数组，线段数的代码量较大，而且代码的运行速度较慢，故如果是线段树与树状数组都可以解的题目，应优先选择树状数组。

比树状数组强大的是，线段树可以求解很多涉及到区间的问题，比如求解区间最大值、最小值等。

线段树的算法图示如下：

 代码实现：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

const int N = 100010;

int n, m;
int w[N];
struct Node{
    int l, r;
    int sum;
}tr[N * 4];

void pushup(int u){
    tr[u].sum = tr[u << 1].sum + tr[u << 1 | 1].sum;
}

void build(int u, int l, int r){
    if(l == r)  tr[u] = {l, r, w[r]};
    else{
        tr[u] = {l, r};
        int mid = l + r >> 1;
        build(u << 1, l, mid), build(u << 1 | 1, mid + 1, r);
        pushup(u);
    }
}

int query(int u, int l, int r){
    if(tr[u].l >= l && tr[u].r <= r)    return tr[u].sum;
    int mid = tr[u].l + tr[u].r >> 1;
    int sum = 0;
    if(l <= mid)    sum = query(u << 1, l, r);
    if(r > mid) sum += query(u << 1 | 1, l, r);
    return sum;
}

void modify(int u, int x, int v){
    if(tr[u].l == tr[u].r)  tr[u].sum += v;
    else{
        int mid = tr[u].l + tr[u].r >> 1;
        if(x <= mid)    modify(u << 1, x, v);
        else    modify(u << 1 | 1, x, v);
        pushup(u);
    }
}

int main(){
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i ++ )   scanf("%d", &w[i]);
    build(1, 1, n);

    while(m -- ){
        int k, a, b;
        scanf("%d%d%d", &k, &a, &b);
        if(k == 0)  cout << query(1, a, b) << endl;
        else modify(1, a, b);
    }

    return 0;
}