线段树lazy标记2

Description
给定一个正整数序列A，要求支持以下操作：
1）：ADD a b c 表示在[a，b]上加上一个常数C。
2）：COVER a b c 把[a，b]整体赋值为一个常数K。
3）：QUERY a b 查询[a，b]的sum。

Input
第一行两个正整数n、m，n表示序列长度，m表示操作数。
第二行n个正整数，第i表示Ai的大小。
接下来的m行，每行有且仅有一种操作，具体和题目描述一致。
n,m<=100000，其他权值都<=50000，小心爆int。

Output
对于每个询问操作，输出答案。

Sample Input
10 10
17 18 16 12 13 7 13 6 11 20
QUERY 5 6
QUERY 2 7
ADD 1 6 13
QUERY 4 10
ADD 2 5 18
COVER 2 8 11
ADD 6 9 5
QUERY 8 8
ADD 6 9 18
QUERY 5 7

Sample Output
20
79
121
16
79

HINT

思路
记录3个标记：icoveri表示这个点是否在下传标记后被覆盖。coveri表示这个点被覆盖成了什么值。sumi表示这个点被增加了多少。下传时先考虑cover的下传，再考虑sum的下传。注意每次一个点被覆盖后这个点的sum值都要被赋成0。

代码

#include <cstdio>const int maxn=100000;long long a[maxn+10];struct segment_tree{  long long val[(maxn<<2)+10],sum[(maxn<<2)+10],cover[(maxn<<2)+10];  int icover[(maxn<<2)+10];  inline int updata(int now)//更新这个点的val  {    val[now]=val[now<<1]+val[now<<1|1];    return val[now];  }  inline int pushdown(int now,int left,int right)//下传标记  {    int mid=(left+right)>>1;    if(icover[now])//如果这个点被cover过，下传cover标记      {        icover[now]=0;        cover[now<<1]=cover[now];        sum[now<<1]=0;//注意sum标记要赋成0        val[now<<1]=cover[now]*(mid-left+1);        cover[now<<1|1]=cover[now];        sum[now<<1|1]=0;        val[now<<1|1]=cover[now]*(right-mid);        icover[now<<1]=icover[now<<1|1]=1;//表示被覆盖过      }    sum[now<<1]+=sum[now];//下传sum标记    val[now<<1]+=sum[now]*(mid-left+1);    sum[now<<1|1]+=sum[now];    val[now<<1|1]+=sum[now]*(right-mid);    sum[now]=0;    return 0;  }  int build(int now,int left,int right)//建树  {    if(left==right)      {        icover[now]=0;//初始化        sum[now]=0;        val[now]=a[left];        return 0;      }    int mid=(left+right)>>1;    build(now<<1,left,mid);    build(now<<1|1,mid+1,right);    updata(now);    return 0;  }  int rcover(int now,int left,int right,int askl,int askr,long long cval)  //将一个区间[left,right]cover成cval  {    if((askl<=left)&&(right<=askr))      {        icover[now]=1;//覆盖        cover[now]=cval;        sum[now]=0;//注意        val[now]=cval*(right-left+1);        return 0;      }    int mid=(left+right)>>1;    pushdown(now,left,right);    if(askl<=mid)      {        rcover(now<<1,left,mid,askl,askr,cval);      }    if(mid<askr)      {        rcover(now<<1|1,mid+1,right,askl,askr,cval);      }    updata(now);    return 0;  }  int add(int now,int left,int right,int askl,int askr,long long cval)  //将一个区间的值增加  {    if((askl<=left)&&(right<=askr))      {        sum[now]+=cval;        val[now]+=cval*(right-left+1);        return 0;      }    int mid=(left+right)>>1;    pushdown(now,left,right);    if(askl<=mid)      {        add(now<<1,left,mid,askl,askr,cval);      }    if(mid<askr)      {        add(now<<1|1,mid+1,right,askl,askr,cval);      }    updata(now);    return 0;  }  long long getsum(int now,int left,int right,int askl,int askr)  //求一段区间的和  {    if((askl<=left)&&(right<=askr))      {        return val[now];      }    int mid=(left+right)>>1;    long long res=0;    pushdown(now,left,right);    if(askl<=mid)      {        res+=getsum(now<<1,left,mid,askl,askr);      }    if(mid<askr)      {        res+=getsum(now<<1|1,mid+1,right,askl,askr);      }    return res;  }};segment_tree st;int n,m,aa,b;long long c;char s[10];int main(){  scanf("%d%d",&n,&m);  for(register int i=1; i<=n; ++i)    {      scanf("%lld",&a[i]);    }  st.build(1,1,n);  while(m--)    {      scanf("%s%d%d",s,&aa,&b);      if(s[0]=='Q')        {          printf("%lld\n",st.getsum(1,1,n,aa,b));        }      else if(s[0]=='C')        {          scanf("%lld",&c);          st.rcover(1,1,n,aa,b,c);        }      else        {          scanf("%lld",&c);          st.add(1,1,n,aa,b,c);        }    }  return 0;}