本题要求用树状数组实现 区间增加 和 单点查询。
前置知识:
题意:
思路:
本题的指令有“区间增加”,“单点查询”。
这两个指令是 裸的差分 所能够完成的,前者O(1)
,后者O(n)
,不过根据题目,元素个数n
和询问次数m
的数据范围均为1e5
,如果用裸的差分,完成m
次单点查询需要O(n^2)
,显然不合法,应当舍弃。
正确做法:树状数组+差分,或者说用差分数组来建立树状数组。
而树状数组基本指令为:“单点增加”,“区间查询”,当结合了差分数组就能够实现“区间增加”,“单点查询”了!
(差分数组的单点增加,对应原数组区间增加,差分数组查询前i
项前缀和对应原数组查询a[i]
)
时间复杂度:
O(mlogn)
代码:
/*将差分数组存入树状数组,原数组即为树状数组的前缀和,将问题转化为了树状数组,
原数组第x个数即为树状数组1~x的前缀和*/
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N = 1e5+10;
int n, m;
int a[N], c[N];
int lowbit(int x) {return x & (-x);}
int ask(int x){
int ans = 0;
while(x) {ans+=c[x], x-=lowbit(x);}
return ans;
}
void add(int x, int y)
{
for(; x<=n; x+=lowbit(x)) c[x]+=y;
}
signed main()
{
cin>>n>>m;
for(int i=1;i<=n;++i)
{
scanf("%lld",&a[i]);
add(i, a[i]-a[i-1]);//用add构造差分树状数组
}
while(m--)
{
char op[2];
int l, r, d, x;
scanf("%s",op);
if(*op=='C') {
scanf("%lld%lld%lld", &l, &r, &d);
add(l, d), add(r+1, -d);//用O(1)的复杂度进行区间修改
}
else{
scanf("%lld", &x);
printf("%lld\n", ask(x));
}
}
return 0;
}