洛谷 P3374 【模板】树状数组 1

给一个数列,进行两种操作:

  1. 某个数 +x+x
  2. 求区间和

和线段树1不同之处在于此题是单点修改,区间查询。

树状数组写起来比线段树简单一点,代码如下:

#include<iostream>
#include<cstdio>
#define MAXN 500010
using namespace std;
int n,m,a[MAXN],b[MAXN],t,x,y;
int lowbit(int x){return x&-x;}
void add(int x,int k){
    while(x<=n){
        b[x]+=k;
        x+=lowbit(x);
    }
}
int sum(int x){
    int res=0;
    while(x>0){
        res+=b[x];
        x-=lowbit(x);
    }
    return res;
}
int query(int x,int y){
    return sum(y)-sum(x-1);
}
int main(){
#ifdef WINE
    freopen("data.in","r",stdin);
#endif
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        add(i,a[i]);
    }
    while(m--){
        scanf("%d%d%d",&t,&x,&y);
        if(t==1)add(x,y);
        if(t==2)printf("%d\n",query(x,y));
    }
    return 0;
}

posted @ 2020-04-04 13:25  winechord  阅读(83)  评论(0编辑  收藏  举报