Description
给定长度为N(N<=105)的数列A,然后输入M行操作指令。
第一类指令形如“C l r d”,表示把数列中第l~r个数都加d。
第二类指令形如“Q X”,表示询问数列中第x个数的值。
对于每个询问,输出一个整数表示答案。
Input
第一行包含两个整数N和M。
第二行包含N个整数A[i]。
接下来M行表示M条指令,每条指令的格式如题目描述所示。
数据范围:
1≤N,M≤10^5,
|d|≤10000,
|A[i]|≤1000000000
Output
对于每个询问,输出一个整数表示答案。
每个答案占一行。
Sample Input
10 5
1 2 3 4 5 6 7 8 9 10
Q 4
Q 1
Q 2
C 1 6 3
Q 2
Sample Output
4
1
2
5
本题的指令有“区间增加”和“单点查询”,而树状数组只支持“单点增加”,需要作出一些转化来解决问题。
新建一个数组b,起初全为0,对于每条指令“C l r d”,我们把它转化为两条指令:
1、把b[l]加上d。
2、再把b[r+1]减去d
执行上面两条指令后,我们来考虑一下b数组的前缀和(b[1~x]的和)的情况:
1、对于1<=x<l,前缀和不变
2、对于l<=x<=r,前缀和增加了d
3、对于r<x<=N,前缀和不变(l处增加d,r+1处减d,抵消)
我们发现,b数组的前缀和b[1~x]就反映了指令“C l r d”对a[x]产生的变化。
用sum[ ]表示b数组的前缀和
sum[1] 表示1个数的和
sum[2] 表示前两个数的和
……
sum[l-1]表示前l-1个数的和
sum[1] 、sum[2] ……sum[l-1]的和都没变
b[l]加了个d
sum[l] 就会增加个d
因为sum[l]增加个d,造成sum[l+1]、sum[l+2]……sum[r]都会增加d
所以b数组从l到r的每个前缀和正好是a数组从l到r的变化量
于是,我们可以用树状数组来维护b的前缀和(对b只有单点增加操作)。因为各次操作之间具有可累加性,所以在树状数组上查询前缀和b[1~x],就得出了到目前为止所有“C”指令在a[x]上增加的数值总和。再加上a[x]的初始值,就得到了Q x的答案。
该做法把“维护b数组的具体值”代替了“维护指令的累积影响”,每次修改在l处增加,在r+1处消除。求b[1-x]前缀和就是a[x]的累积变化量。把“区间增加+单点查询”变化为树状数组最擅长的“单点增加+区间查询”
完整代码:
#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
int a[maxn],c[maxn],l[maxn],r[maxn];
int N,M,n;
inline int lowbits(int x)
{
return x&-x;
}
void add(int x,int y)
{
for(;x<=N;x+=lowbits(x))
c[x]+=y;
}
int ask(int x)
{
int ans=0;
for(;x;x-=lowbits(x))
ans+=c[x];
return ans;
}
int main(){
cin>>N>>M;
for(int i=1;i<=N;i++)
scanf("%d",&a[i]);
memset(c,0,sizeof(c));
while(M--)
{ char k[2];
int k1;
scanf("%s%d",k,&k1);
if(k[0]=='Q')
{
printf("%d\n",a[k1]+ask(k1));
}
if(k[0]=='C')
{
int k2,k3;
scanf("%d%d",&k2,&k3);
add(k1,k3);
add(k2+1,-k3);
}
}
return 0;
}