树状数组(点查+区间插/区间查+点插)模板一样,思想不一样。
点查+区间插:
差分:开两个数组(原始s存区间操作,arr存初始所有数)(arr[b]+count(b)//第b个数)
点插+区间查:
前缀和:开一个数组即可( count(r)-count(l-1);//l至r之和)
本题为点查+区间插
#include<bits/stdc++.h>
using namespace std;
int n,s[100010],arr[100010];
int lowbit(int x) // 返回末尾的1
{
return x & -x;
}
void add(int p,int x) //p位置+x
{
while(p<=n)
{
s[p]+=x;
p+=lowbit(p);
}
}
int count(int p)
{
int res=0;
while(p)
{
res+=s[p];
p-=lowbit(p);
}
return res;
}
int main()
{
int m;
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>arr[i];
}
for(int i=0;i<m;i++)
{
char a;
cin>>a;
int b,c,d;
if(a=='Q') {cin>>b;cout<<arr[b]+count(b)<<endl;}
else
{
cin>>b>>c>>d;
add(b,d),add(c+1,-d);
}
}
}