线段树,一个更复杂的区间修改

  今天AC了一道关于线段树比较复杂的题目,是关于区间修改的,但不同之处在于,修改方式有两种,一种是将给定区间内每个元素加上一个数--add,另一种是将给定区间内的所有元素设置为同一个数--set,对于每一次修改,输出整个区间的和。

  可知本题关键在于pushdown函数,对于线段树中的每个节点,若我们仅仅简单的维护两个懒惰标记,并根据不同的修改方式和标志间的次序进行下放更新的话,显然有很多种情况,并且难以分析,那么根据题目的提示,我们将两个下放操作合并起来,对于线段树中的每一个节点,若我们执行add操作,那么必然需要经所用区间的所有set标志下放,并更新,然后再添加add标志并更新父节点,若我们进行set操作,则只需将所有用到的区间的add标志取消,再将set标志下放更新父节点即可。

  根据以上步骤可知,我们保证每个节点在使用时没有标志,并且整个过程中,每个节点要么没有标志,要么只有一个标志。

#include <iostream>
#include <cstdio>
using namespace std;
long long tree[100010<<2];
int set[100010<<2],add[100010<<2];
void pushup(int rt)
{
	tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}
void pushdown(int l,int r,int rt)
{
	if(l==r) return ;
	int m=(l+r)>>1;
	if(set[rt]!=-1){
		set[rt<<1]=set[rt<<1|1]=set[rt];// 下放
		add[rt<<1]=add[rt<<1|1]=0;// 覆盖
		tree[rt<<1]=(m-l+1)*set[rt];
		tree[rt<<1|1]=(r-m)*set[rt];
		set[rt]=-1;
	}
	if(add[rt]){
		add[rt<<1]+=add[rt];
		add[rt<<1|1]+=add[rt];// 这里的下放必须用+=
		tree[rt<<1]+=(m-l+1)*add[rt];
		tree[rt<<1|1]+=((r-m))*add[rt];
		add[rt]=0;
	}
}
void build(int l,int r,int rt)
{
	set[rt]=-1;add[rt]=0;
	if(l==r){
		scanf("%lld",&tree[rt]);
		return ;
	}
	int m=(l+r)>>1;
	build(l,m,rt<<1);
	build(m+1,r,rt<<1|1);
	pushup(rt);
}
void update(int op,int L,int R,int x,int l,int r,int rt)
{
	pushdown(l,r,rt);// 必须放在开头
	if(L<=l&&r<=R){
		if(op) {set[rt]=x;tree[rt]=(r-l+1)*x;}
		else {add[rt]+=x;tree[rt]+=(r-l+1)*x;}
		return ;
	}
	int m=(l+r)>>1;
	if(L<=m) update(op,L,R,x,l,m,rt<<1);
	if(R>m) update(op,L,R,x,m+1,r,rt<<1|1);
	pushup(rt);
}
int main()
{
	//freopen("in.txt","r",stdin);
	int n,m;
	scanf("%d%d",&n,&m);
	build(1,n+1,1);
	while(m--){
		int op,a,b,c;
		cin>>op>>a>>b>>c;
		update(op,a+1,b+1,c,1,n+1,1);
		printf("%lld\n",tree[1]);
	}
	return 0;
}


 

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值