区间修改时用lazy标记区间平均增加数,同时更新区间和(r-l+1)*ave,查找某点的值的过程中如果某点有lazy值则向下扩散,并把该点lazy值标0。
#include <stdio.h>
const int maxn=1000001*4;
struct node{
int l,r,lazy,sum;
}n[maxn];
void build(int k,int l,int r){
n[k].l=l; n[k].r=r;
if(l==r){
n[k].lazy=0;
n[k].sum=0;
return;
}
int mid=(l+r)>>1;
build(k<<1,l,mid);
build(k<<1|1,mid+1,r);
}
void add(int k,int l,int r,int sum){
if(n[k].l==l&&n[k].r==r){
n[k].lazy+=sum;
n[k].sum+=(r-l+1)*sum;
return;
}
int mid=(n[k].l+n[k].r)>>1;
if(r<=mid) add(k<<1,l,r,sum);
else if(l>mid) add(k<<1|1,l,r,sum);
else{
add(k<<1,l,mid,sum);
add(k<<1|1,mid+1,r,sum);
}
}
int query(int k,int a){
if(n[k].l==n[k].r){
return n[k].sum;
}
if(n[k].lazy){
n[k<<1].lazy+=n[k].lazy;
n[k<<1|1].lazy+=n[k].lazy;
n[k<<1].sum+=(n[k<<1].r-n[k<<1].l+1)*n[k].lazy;
n[k<<1|1].sum+=(n[k<<1|1].r-n[k<<1|1].l+1)*n[k].lazy;
n[k].lazy=0;
}
int mid=(n[k].l+n[k].r)>>1;
if(a<=mid) query(k<<1,a);
else query(k<<1|1,a);
}
int main(){
int t,m,a,b,c;
char s[7];
scanf("%d %d",&t,&m);
build(1,1,m);
while(t--){
scanf("%s",s);
if(s[0]=='A'){
scanf("%d %d %d",&a,&b,&c);
add(1,a,b,c);
}
else{
scanf("%d",&a);
printf("%d\n",query(1,a));
}
}
}
区间修改,单点查询,query()查询的为单点值。
区间查找,单点修改,query()查询的为前缀和。
#include <stdio.h>
int l[1000005],m;
int lowbit(int x){
return x&(-x);
}
void add(int x,int y){
while(x<=m){
l[x]+=y;
x+=lowbit(x);
}
}
int query(int x){
int sum=0;
while(x>0){
sum+=l[x];
x-=lowbit(x);
}
return sum;
}
int main(){
int t,a,b,c;
char s[7];
scanf("%d %d",&t,&m);
while(t--){
scanf("%s",s);
if(s[0]=='A'){
scanf("%d %d %d",&a,&b,&c);
add(a,c);add(b+1,-c);
}
else{
scanf("%d",&a);
printf("%d\n",query(a));
}
}
}