树状数组相关 模板

树状数组区间和单点修改查询的模板

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6+10;
int a[maxn],n,x,y,k,sum;
inline int lowbit(int x) {return x&-x;}
//单点修改 区间查询
void update(int x,int k){ for (int i = x;i<=n;i+=lowbit(i)) a[i] += k;}
int query(int x){
    int res = 0;
    for (int i=x;i>0;i-=lowbit(i)) res += a[i];
    return res;
}
//区间修改单点查询 update时传入a[i]-a[i-1] 差分建树 query时即查询单点值
//区间修改区间查询 建两个树状数组
int sum1[maxn],sum2[maxn];
void Update(int x,int k) {for (int i=x;i<=n;i+=lowbit(i)) sum1[i] += k,sum2[i] += k * (x - 1);}
int Query(int x){
    int res = 0;
    for (int i = x;i;i-=lowbit(i)) res += x * sum1[i] - sum2[i];
    return res;
}
int main (){
    cin>>n;

    //单点修改 区间查询
    for (int i=1;i<=n;++i) {
        cin>>a[i];
        update(i,a[i]); //单点修改
    }
    cin>>x;
    sum = query(x);//查询区间 1~x 的和



    //区间修改 单点查询
    for (int i=1;i<=n;++i){
        cin>>a[i];
        update(i,a[i]-a[i-1]); //差分建树
    }
    cin>>x>>y;
    //区间[x,y]内每个数加k 区间内两数之差依然不变 即修改两个点
    update(x,k);
    update(y+1,-k);
    cin>>x;
    sum = query(x);//查询x点的值



    //区间修改区间查询
    for (int i=1;i<=n;++i){
        cin>>a[i];
        Update(i,a[i]-a[i-1]);
    }
    //[x,y]区间内加上k
    Update(x,k);    //A[x] - A[x-1]增加k
    Update(y+1,-k);        //A[y+1] - A[y]减少k
    //求[x,y]区间和
    sum = Query(y) - Query(x-1);

    return 0;
}