[POJ 3468] A Simple Problem with Integers [树状数组]

给一列数，有区间修改区间查询，线段树模板题。

使用树状数组的区间修改区间查询版本..

设原数组为a[]，现记录b[i]为原数组中a[1]...a[i]整体被加了多少，c[i]为原书组中a[1]...a[i]整体被加了多少的和，即c[i]=b[i]*i。

进行区间修改[l,r]时，只需让b[r]+=x，b[l-1]-=x即可，对应的c[i]的变化量为b[i]*i。

进行区间查询[1,l]时，只需计算c[1]+c[2]+...+c[l]+(b[l+1]+b[l+2]+...+b
)*l即可。

#include <cstring>
#include <cstdio>

struct BIT {
int n;
long long b[100002];
long long c[100002];
void clear(int nn) {
n=nn;
for (int i=0;i<=n;i++) b[i]=c[i]=0;
}
inline int lb(int i) {
return i&-i;
}
void set(long long a[],int i,long long x) {
if (i==0) return;
for (;i<=n;i+=lb(i)) a[i]+=x;
}
long long get(long long a[],int i) {
long long ans=0;
for (;i>0;i-=lb(i)) ans+=a[i];
return ans;
}
void set(int x,int y,long long z) {
set(b,y,z);
set(b,x-1,-z);
set(c,y,z*y);
set(c,x-1,-z*(x-1));
}
long long get(int i) {
return get(c,i)+(get(b,n)-get(b,i))*i;
}
};

int n,q;
BIT c;

int main() {
int i,x,y,z;
char cc;
scanf("%d%d",&n,&q);
c.clear(n);
for (i=1;i<=n;i++) {
scanf("%d",&x);
c.set(i,i,x);
}
while (q--) {
scanf(" %c",&cc);
if (cc=='C') {
scanf("%d%d%d",&x,&y,&z);
c.set(x,y,z);
} else {
scanf("%d%d",&x,&y);
printf("%lld\n",c.get(y)-c.get(x-1));
}
}
return 0;
}