【线段树】区间求和+单点修改

#include
#include

using namespace std;

int n,m,a[100004];
struct data{int l,r,val;}tr[2*100004];

void build(int k,int s,int t) {//建树
    tr[k].l=s;tr[k].r=t;
    if(s==t) {tr[k].val=a[s];return;}
    int mid=(s+t)>>1;
    build(k<<1,s,mid);
    build(k<<1|1,mid+1,t);
    tr[k].val=tr[k<<1].val+tr[k<<1|1].val;
}

int ask1(int k,int x,int y){//区间求和
    int ll=tr[k].l,rr=tr[k].r;
    if(x==ll && y==rr) return tr[k].val;
    int mid=(ll+rr)>>1; 
    if(y<=mid) return ask1(k<<1,x,y);
    else if(x>mid) return ask1(k<<1|1,x,y);
    else return ask1(k<<1,x,mid)+ask1(k<<1|1,mid+1,y);
}

void add(int k,int x,int addval){//单点更新
    tr[k].val+=addval;
    int ll=tr[k].l,rr=tr[k].r;
    if(ll==rr) return;
    int mid=(ll+rr)>>1;
    if(x<=mid) add(k<<1,x,addval);
    else add(k<<1|1,x,addval);
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    build(1,1,n);
    for(int i=1;i<=m;i++){
        int flag;scanf("%d",&flag);
        int x,y;scanf("%d%d",&x,&y);
        if(flag==1) add(1,x,y);
        else printf("%d\n",ask1(1,x,y));
    }
    return 0;
}

样例输入:
6 4
4
5
6
2
1
3
1 3 5
2 1 4
1 1 9
2 2 6

样例输出:
22
22

你可能感兴趣的:(知识/模板)