给定长度为 N 的数列 A,然后输入 M 行操作指令。
第一类指令形如 C l r d
,表示把数列中第 l∼r 个数都加 d。
第二类指令形如 Q x
,表示询问数列中第 x 个数的值。
对于每个询问,输出一个整数表示答案。
输入格式
第一行包含两个整数 N 和 M。
第二行包含 N 个整数 A[i]。
接下来 M 行表示 M 条指令,每条指令的格式如题目描述所示。
输出格式
对于每个询问,输出一个整数表示答案。
每个答案占一行。
数据范围
1≤N,M≤105,
|d|≤10000,
|A[i]|≤109
输入样例:
10 5
1 2 3 4 5 6 7 8 9 10
Q 4
Q 1
Q 2
C 1 6 3
Q 2
输出样例:
4
1
2
5
分析:正常的树状数组应该是一次修改一个值,查询时查询一个区间,本题是修改一个区间,查询一个值,为了还能用树状数组,我们可以用树状数组维护一个差分数组,这样就可以把区间修改变成单个值的修改,单个查询就变成了区间查询。
代码如下:
#include <iostream>
using namespace std;
const int N=1e5+10;
int t[N],a[N];
int n,m,l,r,d;
void add(int x,int w)
{
for(;x<=n;x+=(x&-x)) t[x]+=w;
}
int getsum(int x)
{
int res=0;
for(;x;x-=(x&-x)) res+=t[x];
return res;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>a[i];
add(i,a[i]-a[i-1]);
}
while(m--)
{
char q;
cin>>q;
if(q=='C')
{
cin>>l>>r>>d;
add(l,d);
add(r+1,-d);
}
else
{
cin>>d;
cout<<getsum(d)<<endl;
}
}
return 0;
}