树状数组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;
}