线段树(该段求段+lazy)优化,洛谷之提高历练地,提高模板-nlogn数据结构

正文

      这个东西挺简单的吧,线段树就不细讲了,主要讲讲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

你可能感兴趣的:(线段树(该段求段+lazy)优化,洛谷之提高历练地,提高模板-nlogn数据结构)