CSU OJ2151 集训难度

小L正在组织acm暑假集训，但众所周知，暑假集训的萌新中有OI神犇，也有暑假才开始学算法的萌新，如果统一集训的难度，无法很好地让萌新们得到训练，所以小L想了一个办法，根据每次测试的情况，改变萌新们的集训难度。现在将萌新们编号为1到n，最初萌新们的集训难度为v0，测试后有两种操作，第一种是某一区间的萌新的集训难度同时提高，另一种是将某一段区间的萌新的集训难度变为同一个数，同时，Wells希望在某次调整难度之后，知道某一段区间的萌新的集训难度之和，由于小L比较鶸，他并不知道如何快速解决这个问题，你能帮帮他嘛？

Input

第一行三个数n,m,v0 表示有n名萌新和m次调整，初始时全部萌新的集训难度都为v0

第2~m+1行 每行三个数或四个数

0 x y v 表示把 [x,y]区间内的萌新的集训难度都增加v

1 x y v 表示把 [x,y]区间内的萌新的集训难度都变为v

2 x y表示询问[x,y]区间内萌新的集训难度之和

0<n,m<=10^5， |v|<=10^5

Output

每个询问一行，输出答案

Sample Input

3 5 0
0 1 3 1
1 2 3 2
2 1 1
2 2 2
2 2 3

Sample Output

1
2
4

Hint

题解：线段树基本操作，区间加上一个数，区间变为一个数，区间和；

 #include<cstdio>
 #include<algorithm>
 #define N 100001
 using namespace std;
 int n,m,x,y;
 long long z,val,ans;
 struct node
 {
     int l,r,siz;
     long long set,add,sum;
     bool v;
 } tree[N<<]; 

 void build(int k,int l,int r)
 {
     tree[k].l=l; tree[k].r=r; tree[k].siz=r-l+;
     if(l==r)
     {
         tree[k].sum=val;
         return ;
     }
     int mid=l+r>>;
     build(k<<,l,mid);
     build(k<<|,mid+,r);
     tree[k].sum=tree[k<<].sum+tree[k<<|].sum;
 }

 void pushup(int k)
 {
     tree[k].sum=tree[k<<].sum+tree[k<<|].sum;
 }

 void pushdown_set(int k)
 {
     tree[k<<].add=tree[k<<|].add=;
     tree[k<<].set=tree[k<<|].set=tree[k].set;
     tree[k<<].v=tree[k<<|].v=true;
     tree[k<<].sum=tree[k<<].siz*tree[k].set;
     tree[k<<|].sum=tree[k<<|].siz*tree[k].set;
     tree[k].v=false; tree[k].set=;
 }

 void pushdown_add(int k)
 {
     tree[k<<].sum+=tree[k<<].siz*tree[k].add;
     tree[k<<|].sum+=tree[k<<|].siz*tree[k].add;
     tree[k<<].add+=tree[k].add;
     tree[k<<|].add+=tree[k].add;
     tree[k].add=;
 }

 void Set(int k,int l,int r,long long z)//区间修改为某个值
 {
     if(tree[k].l>=l&&tree[k].r<=r)
     {
         tree[k].set=z; tree[k].v=true;
         tree[k].sum=z*tree[k].siz;
         tree[k].add=;
         return;
     }
     if(tree[k].v) pushdown_set(k);
     if(tree[k].add) pushdown_add(k);
     int mid=tree[k].l+tree[k].r>>;
     if(l<=mid) Set(k<<,l,r,z);
     if(r>mid) Set(k<<|,l,r,z);
     pushup(k);
 }

 void Add(int k,int l,int r,long long z)//区间加上某个值
 {
     if(tree[k].l>=l&&tree[k].r<=r)
     {
         tree[k].add+=z;
         tree[k].sum+=z*tree[k].siz;
         return;
     }
     if(tree[k].v) pushdown_set(k);
     if(tree[k].add) pushdown_add(k);
     int mid=tree[k].l+tree[k].r>>;
     if(l<=mid) Add(k<<,l,r,z);
     if(r>mid) Add(k<<|,l,r,z);
     pushup(k);
 } 

 void query(int k,int l,int r)
 {
     if(tree[k].l>=l&&tree[k].r<=r)
     {
         ans+=tree[k].sum;
         return;
     }
     if(tree[k].v) pushdown_set(k);
     if(tree[k].add) pushdown_add(k);
     int mid=tree[k].l+tree[k].r>>;
     if(l<=mid) query(k<<,l,r);
     if(r>mid) query(k<<|,l,r);
 }
 int main()
 {
     scanf("%d%d%lld",&n,&m,&val);
     build(,,n);
     while(m--)
     {
         int temp;
         scanf("%d",&temp);
         if(temp==)
         {
             scanf("%d%d%lld",&x,&y,&z);
             Add(,x,y,z);
         }
         else if(temp==)
         {
             scanf("%d%d%lld",&x,&y,&z);
             Set(,x,y,z);
         }
         else
         {
             scanf("%d%d",&x,&y);
             ans=;
             query(,x,y);
             printf("%lld\n",ans);
         }

     }
 }

参考代码：