洛谷 P3368 【模板】树状数组 2

给一个数列,进行以下操作:

  1. 某区间每个数 +x+x
  2. 求某位置的值

区间修改,单点查询。用数组 b 来存放修改产生的差分。

代码如下:

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

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