树状数组 区间修改 差分

传送门

// by spli
#include
#include
#include
#include
using namespace std;

const int N=500010;
int n,m;
int a[N],c[N];

int lowbit(int x){return x&(-x);}

void add(int pos,int v){
    for(int i=pos;i<=n;i+=lowbit(i))
        c[i]+=v;
}

int query(int pos){
    int ret=0;
    for(int i=pos;i>=1;i-=lowbit(i))
        ret+=c[i];
    return ret;
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i) scanf("%d",&a[i]);
    add(1,a[1]);
    for(int i=2;i<=n;++i) add(i,a[i]-a[i-1]);
    int op,x,y,z;
    while(m--){
        scanf("%d",&op);
        if(op==1){
            scanf("%d%d%d",&x,&y,&z);
            add(x,z);
            add(y+1,-z);
        }
        else{
            scanf("%d",&x);
            printf("%d\n",query(x));
        }
    }
    return 0;
}

你可能感兴趣的:(模板,洛谷,树状数组)