树状数组区间和单点修改查询的模板
#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;
}