洛谷P3372--线段树代码模板1

如题，已知一个数列，你需要进行下面两种操作：

1.将某区间每一个数加上x

2.求出某区间每一个数的和

输入格式

第一行包含两个整数N、M，分别表示该数列数字的个数和操作的总个数。

第二行包含N个用空格分隔的整数，其中第i个数字表示数列第i项的初始值。

接下来M行每行包含3或4个整数，表示一个操作，具体如下：

操作1： 格式：1 x y k 含义：将区间[x,y]内每个数加上k

操作2： 格式：2 x y 含义：输出区间[x,y]内每个数的和

输出格式

输出包含若干行整数，即为所有操作2的结果。

输入输出样例

输入 #1复制

5 5
1 5 4 2 3
2 2 4
1 2 3 2
2 3 4
1 1 5 1
2 1 4

输出 #1复制

11
8
20

说明/提示

时空限制：1000ms,128M

数据规模：

对于30%的数据：N<=8，M<=10

对于70%的数据：N<=1000，M<=10000

对于100%的数据：N<=100000，M<=100000

（数据已经过加强^_^，保证在int64/long long数据范围内）

样例说明：

 #include<iostream>
 #include<stdio.h>
 #include<cstring>
 #include<cmath>
 #include<vector>
 #include<stack>
 #include<map>
 #include<set>
 #include<list>
 #include<queue>
 #include<string>
 #include<algorithm>
 #include<iomanip>
 using namespace std;
 const int maxn = ;
 int arr[maxn +];
 struct node
 {
     /* data */
     long long  value;//该结点维护的值
     int left;
     int right;
     long long add;//lazy 每个区间的增量
 }tree[maxn*  + ];

 inline void PushUp(int root)
 {
     tree[root].value = tree[root <<].value + tree[root<<|].value;
 }

 void Build(int root,int x,int y)
 {
     tree[root].left = x;
     tree[root].right = y;
     if(x == y)
     {
         tree[root].value = arr[x];
         return ;
     }
     int mid = (x + y) >> ;
     Build(root<<,      x,mid);
     Build(root<<|,mid+,y);
     PushUp(root);
 }

 void Spread(int root)//懒标记
 {
     if(tree[root].add)//如果树根结点的懒标记不为0 ，修改左右儿子的值
     {
         tree[root<<].value   += tree[root].add * (tree[root<<].right   - tree[root<<].left   +);
         tree[root<<|].value += tree[root].add * (tree[root<<|].right - tree[root<<|].left +);
         //左右儿子的值增加为父亲结点的增量乘以自己各区间的长度
         tree[root<<].add   += tree[root].add;
         tree[root<<|].add += tree[root].add;
         tree[root].add = ;//父亲结点增量变为0;
     }
 }
 void Change(int root,int x,int y,int z)//给根结点为root 区间为(x,y)的每个值加上z
 {
     if( x <= tree[root].left && y  >= tree[root].right)//区间被完全覆盖
     {
         tree[root].value += (long long )z * (tree[root].right - tree[root].left + );
         tree[root].add   += z;
         return ;
     }
     Spread(root);//如果没有发现区间被覆盖，即需要继续向下找，并把懒标记下放
     int mid = (tree[root].right + tree[root].left) >> ;
     if( x <= mid)//要修还区间覆盖了左儿子，修改
     {
         Change(root<<,x,y,z);
     }
     if( y > mid)//右儿子同理
     {
         Change(root<<|,x,y,z);
     }
     PushUp(root);
 }

 long long Query(int root,int x,int y)//对区间(x,y)的查询操作
 {
     long long ans = ;
     if( x <= tree[root].left && y >= tree[root].right )//查询区间被根结点所对应区间覆盖
     {
         return tree[root].value;
     }
     Spread(root);//下传懒标记
     int mid = (tree[root].left + tree[root].right) >> ;
     if ( x <= mid)
     {
         ans += Query(root<<,x,y);
     }
     if( y > mid)
     {
         ans += Query(root<<|,x,y);
     }
     return ans;
 }

 int main()
 {
     int n,m;
     scanf("%d%d",&n,&m);
         for(int i = ; i<= n;i++)
         {
             scanf("%d",&arr[i]);
         }
         Build(,,n);
         for(int i =  ;i < m;i++)
         {
             int flag;
             int x,y,k;
             cin>>flag;
             if(flag == )
             {
                 scanf("%d%d%d",&x,&y,&k);
                 Change(,x,y,k);
             }
             else
             {
                 scanf("%d%d",&x,&y);
                 cout<<Query(,x,y)<<endl;
             }
         }
     return ;
 }