線段樹lazy標記2

描述

給定一個正整數序列A，要求支援以下操作
1）： ADD a b c 表示在[a，b]上加上一個常數C。
2）： COVER a b c 把[a，b]整體賦值為一個常數K。
3）： QUERY a b 查詢[a，b]的sum。

輸入

第一行兩個正整數n、m，n表示序列長度，m表示運算元
第二行n個正整數，第i表示A[i]的大小
接下來的m行，每行有且僅有一種操作，具體和題目描述一致
n,m<=100000
其他權值都<=50000
小心爆int

輸出

對於每個詢問操作，輸出答案

樣例

輸入

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

輸出

20
79
121
16
79

  1 #include<bits/stdc++.h>
  2 #define ll long long
  3 using namespace std;
  4 struct data{
  5     ll add,sum,l,r,cov;
  6     #define l(x) tree[x].l
  7     #define r(x) tree[x].r
  8     #define sum(x) tree[x].sum
  9     #define add(x) tree[x].add
 10     #define cov(x) tree[x].cov
 11
 
 }tree[4*100001];
 12 ll n,m,a[100001];
 13 void build(ll p,ll l,ll r)
 14 {
 15     l(p)=l;r(p)=r;
 16     if(l==r)
 17     {
 18         sum(p)=a[l];
 19         return;
 20     }
 21     ll mid=(l+r)>>1;
 22     build(p<<1,l,mid);
 23     build(p<<1|1,mid+1,r);
 24     sum(p)=sum(p<<1
 
)+sum(p<<1|1);
 25 }
 26 void push_down(ll p)
 27 {
 28     if(cov(p))
 29     {
 30         sum(p<<1)=cov(p)*(r(p<<1)-l(p<<1)+1);
 31         sum(p<<1|1)=cov(p)*(r(p<<1|1)-l(p<<1|1)+1);
 32         cov(p<<1)=cov(p);
 33         cov(p<<1|1)=cov(p);
 34         cov(p)=0;
 35         add(p<<1)=0;
 36         add(p<<1|1)=0;
 37     }
 38     if(add(p))
 39     {
 40         sum(p<<1)+=add(p)*(r(p<<1)-l(p<<1)+1);
 41         sum(p<<1|1)+=add(p)*(r(p<<1|1)-l(p<<1|1)+1);
 42         add(p<<1)+=add(p);
 43         add(p<<1|1)+=add(p);
 44         add(p)=0;
 45     }
 46 }
 47 void check(ll p,ll l,ll r,ll v)
 48 {
 49     if(l<=l(p)&&r>=r(p))
 50     {
 51         sum(p)+=v*(r(p)-l(p)+1);
 52         add(p)+=v;
 53         return;
 54     }
 55     push_down(p);
 56     ll mid=(l(p)+r(p))>>1;
 57     if(l<=mid)
 58     check(p<<1,l,r,v);
 59     if(r>mid)
 60     check(p<<1|1,l,r,v);
 61     sum(p)=sum(p<<1)+sum(p<<1|1);
 62 }
 63 void check2(ll p,ll l,ll r,ll v)
 64 {
 65     if(l<=l(p)&&r>=r(p))
 66     {
 67         sum(p)=v*(r(p)-l(p)+1);
 68         add(p)=0;
 69                 cov(p)=v;
 70         return;
 71     }
 72     push_down(p);
 73     ll mid=(l(p)+r(p))>>1;
 74     if(l<=mid)
 75     check2(p<<1,l,r,v);
 76     if(r>mid)
 77     check2(p<<1|1,l,r,v);
 78     sum(p)=sum(p<<1)+sum(p<<1|1);
 79 }
 80 ll ask(ll p,ll l,ll r)
 81 {
 82     if(l<=l(p)&&r>=r(p))
 83     return sum(p);
 84     push_down(p);
 85     ll val=0;
 86     ll mid=(l(p)+r(p))>>1;
 87     if(l<=mid)
 88     val+=ask(p<<1,l,r);
 89     if(r>mid)
 90     val+=ask(p<<1|1,l,r);
 91     return val;
 92 }
 93 int main()
 94 {
 95     scanf("%lld%lld",&n,&m);
 96     for(ll i=1;i<=n;i++)
 97     scanf("%lld",&a[i]);
 98     build(1,1,n);
 99     for(ll i=1;i<=m;i++)
100     {
101         int x,y,k;
102         char c[10];
103         scanf("%s%d%d",c,&x,&y);
104         if(c[0]=='A')
105         {
106             scanf("%d",&k);
107             check(1,x,y,k);
108         }
109         else if(c[0]=='C')
110         {
111             scanf("%d",&k);
112             check2(1,x,y,k);
113         }
114         else if(c[0]=='Q')
115             printf("%lld\n",ask(1,x,y));
116     }
117     return 0;
118 }