线段树模板题——动态求连续区间和（《信息学奥赛一本通》）

给定 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,
数据保证在任何时候，数列中所有元素之和均在 int 范围内。

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

using namespace std;
const int N = 100005;
int n,m;
int w[N];
struct node
{
    int l,r,sum;
}tr[N*4];
void push_up(int u)
{
    //左儿子u*2 右儿子u*2+1
    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);
        push_up(u);
    }
}
int query(int u,int l,int r)
{
    //区间被完全包含的话，直接返回区间的值
    if(l<=tr[u].l && tr[u].r <=r) return tr[u].sum;
    int mid = tr[u].l + tr[u].r >> 1;
    int sum = 0;
    if(mid >= l) sum += query(u<<1,l,r);  //区间中点和左边有没有交集
    if(r>=mid+1) sum += query(u<<1|1,l,r);
    return sum;
}
void modify(int u,int x,int v) //修改某个u结点以及，修改位置为x，修改值为+=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);
        
        push_up(u);
    }
}
int main()
{
    scanf("%d%d", &n, &m);
    for(int i=1; i<=n; ++i){
		scanf("%d",&w[i]);
	}
	build(1,1,n);  //根节点下标1，初始区间[1,n]
    for(int i=0; i<m; ++i){
        int k,ta,tb;
        scanf("%d%d%d", &k, &ta, &tb);
        if(k==0){
            //传根节点的编号，和查询区间
            printf("%d\n",query(1,ta,tb));
        }
        else{
            modify(1,ta,tb); //传根节点下标，修改位置和修改的值
        }
    }
    return 0;
}