线段树的区间操作(区间更新,区间查询)

区间操作
struct node {
	int l,r,sum,lazy;
} tr[maxn<<2];   //节点类型
void pushup(int m) {   //某个值的向上更新
	tr[m].sum=tr[m<<1].sum+tr[m<<1|1].sum;
}
void build(int m,int l,int r) {//建树
	tr[m].l=l;
	tr[m].r=r;
	tr[m].lazy=0;            //lazy初始化为0
	if(l==r) {
		tr[m].sum=0;
		return ;
	}           //如果只有一个节点
	int mid=(l+r)>>1;
	build(m<<1,l,mid);
	build(m<<1|1,mid+1,r);           //建好左右半边
	pushup(m);//向上更新
}
void pushdown(int m) {//因为有个lazy 值,向下更新
	if(tr[m].lazy) {//如果有lazy 值 
		tr[m<<1].lazy+=tr[m].lazy;
		tr[m<<1|1].lazy+=tr[m].lazy;//m的左节点和右节点都加上m节点的lazy 值 
		tr[m<<1].sum+=tr[m].lazy*(tr[m<<1].r-tr[m<<1].l+1);//左半边区间都加上lazy 值 
		tr[m<<1|1].sum+=tr[m].lazy*(tr[m<<1|1].r-tr[m<<1|1].l+1);//右半边区间都加上lazy 值 
		tr[m].lazy=0;//m节点处的lazy 已经处理完毕,可以置为0(因为已经处理了) 
	}
}
void updata(int m,int l,int r,int val) {//区间更新
	if(tr[m].l==l&&tr[m].r==r) {
		tr[m].sum+=val*(r-l+1);//算出这个区间的新sum
		tr[m].lazy+=val;//更新lazy 值
		return ;
	}//正好是要更新的区间
	pushdown(m);//向下更新
	int mid=(tr[m].l+tr[m].r)>>1;//中间值
	if(r<=mid)
		updata(m<<1,l,r,val);//更新左边区间
	else if(l>mid)
		updata(m<<1|1,l,r,val);//更新右边区间
	else {//mid 在所要更新的区间里边
		updata(m<<1,l,mid,val);//更新(l,mid)
		updata(m<<1|1,mid+1,r,val);//更新(min+1,r)
	}
	pushup(m);//向上更新
}
int query(int m,int l,int r) {//遍历 
	if(tr[m].l==l&&tr[m].r==r) {
		return tr[m].sum;
	}//返回所求区间的sum 
	pushdown(m);//向下更新 
	int mid=(tr[m].l+tr[m].r)>>1;
	int temp;
	if(r<=mid)
		temp=query(m<<1,l,r);//遍历左半边 
	else if(l>mid)
		temp=query(m<<1|1,l,r);//遍历右半边 
	else
		temp=query(m<<1,l,mid)+query(m<<1|1,mid+1,r);//遍历(l,mid),(mid+1,r) 
	return temp;
}


 

你可能感兴趣的:(日常刷题)