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

  今天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;}