树状数组2【拆分+求前缀和】

树状数组2【拆分+求前缀和】

洛谷P3368

#include<bits/stdc++.h>
using namespace std;

#define int long long
const int N = 500000;

int n,m,d,x,y,z;
int a[N + 5],tree[N + 5];           //分别用来存储原始序列和树状数组

void update(int x,int y){           //更新树状数组
    for(;x <= n;x += x & -x)
        tree[x] += y;
}
int get_sum(int x){                 //求前缀和
    int ans = 0;
    for(;x > 0;x -= x & -x)
        ans += tree[x];
    return ans;
}

signed main(){
    //数组：a1,a2,a3,a4,a5
    //将其进行拆分：a1,a2-a1,a3-a2,a4-a3,a5-a4
    //易得：原数组是拆分数组的前缀和
    //将拆分后的数组作为叶节点从而建立树状数组
    //这样我们如果要求某一个位置的值，就是树状数组前缀和了
    //这样操作还可以使得区间更新更加方便，比如要将[2,3]区间都加上x
    //那么只需要在拆分数组的2位置加上x（2及其之后每一位都加上了x）
    //同时在位置4减去x（4及其之后的每一位都减去x，和前面加上的x进行抵消）
    cin>>n>>m;
    for(int i = 1;i <= n;++i){          //建立树状数组
        cin>>a[i];
        update(i,a[i] - a[i - 1]);
    }
    while(m--){
        cin>>d;
        if(d == 1){
            cin>>x>>y>>z;
            update(x,z);
            update(y + 1,-z);
        }
        else{
            cin>>x;
            cout<<get_sum(x)<<"\n";
        }
    }
    return 0;
}