正文
这个东西挺简单的吧,线段树就不细讲了,主要讲讲lazy。
懒嘛~
如果当前覆盖整个区间,那么我们就用lazy把它那个值记录下来,然后,如果不是完全覆盖当前区间,那么我们就把它下放,乘的话那你就用一下乘法分配律搞一下即可
#include
#include
#include
int n,m;
struct tree{int x,y,ls,rs;long long c,lazy;};
tree tr[200010];
int len=0;
void bt(int x,int y){
len++;
int i=len;
tr[i].x=x;tr[i].y=y;
tr[i].ls=-1;tr[i].rs=-1;
tr[i].c=0;
tr[i].lazy=0;
if(x==y) return ;
int mid=(x+y)/2;
tr[i].ls=len+1;bt(x,mid);
tr[i].rs=len+1;bt(mid+1,y);
}
void add(int p,int x,int y,int t){
if(tr[p].x==x && tr[p].y==y){
tr[p].lazy+=t;
tr[p].c+=t*(y-x+1);
return ;
}
int ls=tr[p].ls,rs=tr[p].rs;
if(tr[p].lazy!=0){
tr[ls].c+=tr[p].lazy*(tr[ls].y-tr[ls].x+1);
tr[ls].lazy+=tr[p].lazy;
tr[rs].c+=tr[p].lazy*(tr[rs].y-tr[rs].x+1);
tr[rs].lazy+=tr[p].lazy;
tr[p].lazy=0;
}
int mid=tr[tr[p].ls].y;
if(y<=mid) add(tr[p].ls,x,y,t);
else if(x>mid) add(tr[p].rs,x,y,t);
else add(tr[p].ls,x,mid,t),add(tr[p].rs,mid+1,y,t);
tr[p].c=tr[tr[p].ls].c+tr[tr[p].rs].c;
}
long long get_total(int p,int x,int y){
if(tr[p].x==x && tr[p].y==y)
return tr[p].c;
int ls=tr[p].ls,rs=tr[p].rs;
if(tr[p].lazy!=0){
tr[ls].c+=tr[p].lazy*(tr[ls].y-tr[ls].x+1);
tr[ls].lazy+=tr[p].lazy;
tr[rs].c+=tr[p].l azy*(tr[rs].y-tr[rs].x+1);
tr[rs].lazy+=tr[p].lazy;
tr[p].lazy=0;
}
int mid=tr[ls].y;
if(y<=mid) return get_total(ls,x,y);
else if(x>mid) return get_total(rs,x,y);
else return get_total(ls,x,mid)+get_total(rs,mid+1,y);
}
int main(){
scanf("%d %d",&n,&m);
bt(1,n);
for(int i=1;i<=n;i++){
int x;
scanf("%d",&x);
add(1,i,i,x);
}
for(int i=1;i<=m;i++){
int t,x,y;
scanf("%d %d %d",&t,&x,&y);
if(t==1) {
int op;
scanf("%d",&op);
add(1,x,y,op);
}
else printf("%lld\n",get_total(1,x,y));
}
}
乘法的话pushdown搞一搞
#include
#include
#define ll long long
struct node{int l,r,lc,rc;ll c,lza,lzm;}tr[400010];
ll a[100010];
int n,m,len=0;
ll p;
void bt(int l,int r)
{
len++;int now=len;
tr[now].l=l;tr[now].r=r;tr[now].lc=tr[now].rc=-1;tr[now].c=tr[now].lza=0;tr[now].lzm=1;
if(l