给定长度为N的数列A,然后输入M行操作指令。
第一类指令形如“C l r d”,表示把数列中第l~r个数都加d。
第二类指令形如“Q X”,表示询问数列中第x个数的值。
对于每个询问,输出一个整数表示答案。
输入格式
第一行包含两个整数N和M。
第二行包含N个整数A[i]。
接下来M行表示M条指令,每条指令的格式如题目描述所示。
输出格式
对于每个询问,输出一个整数表示答案。
每个答案占一行。
数据范围
1≤N,M≤1051≤N,M≤105,
|d|≤10000|d|≤10000,
|A[i]|≤1000000000|A[i]|≤1000000000
输入样例:
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
思路:
树状数组可以把某一个数+x,求一段区间的和
而这道题是让我们在某一个区间加x,求一个数的值
所以这道题可以采取树状数组与差分结合的思路,树状数组存储的是原数组的差分值。
同时我们观察到数据范围较大,所以用long long来得到答案
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=100010;
int n,m;
typedef long long LL;
int a[N];
LL tr[N];
int lowbit(int x){
return x&-x;
}
void add(int x,int c){
for(int i=x;i<n;i+=lowbit(i)) tr[i]+=c;
}
LL sum(int x){
LL res=0;
for(int i=x;i;i-=lowbit(i)) res+=tr[i];
return res;
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=1;i<=n;i++) add(i,a[i]-a[i-1]);//加进去的是a[i]的差分数组,方便进行后面的操作
while(m--){
char op[2];
int l,r,d;
scanf("%s%d",op,&l);
if(*op=='C'){
scanf("%d%d",&r,&d);
add(l,d);
add(r+1,-d);
}
else{
printf("%lld\n",sum(l));
}
}
return 0;
}