洛谷 P3372 【模板】线段树 加法

题目描述

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

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数据范围内）

AC代码:

 #include<iostream>
 using namespace std;
 long long n,m,a[],p,x,y,pp;
 struct kkk{
     int l,r;//l、r分别代表此节点所含信息区间左端和右端
     long long add,tql;//add是lazy标记，tql代表此节点所含区间和。
 }t[];
 void build(int q,int w,int e) {
     t[q].l = w;t[q].r = e;
     if(w == e) { //如果只维护一个节点的信息，说明为叶节点
         t[q].tql = a[w];
         return ;
     }
     int mid = w + e >> ;
     build(*q, w, mid);     //建左子树
     build(*q+, mid+, e); //建右子树
     t[q].tql = t[q*].tql + t[q*+].tql; //一个节点的信息由它的左右儿子得来
 }
 void spr(int o) {//运用lazy标记进行区间修改
     if(t[o].add) {
         t[*o].tql += t[o].add * (t[*o].r - t[*o].l + );
         t[o*+].tql += t[o].add * (t[o*+].r - t[o*+].l + );
         t[o*].add += t[o].add;
         t[o*+].add += t[o].add;
         t[o].add = ;
     }
 }
 void pl(int o,int l,int r,int k) {
     if(l <= t[o].l && r >= t[o].r) {//如果已知区间被所求全部包含，那么直接计算即可
         t[o].tql += (long long) k * (t[o].r - t[o].l + );
         t[o].add += k;
         return ;
     }
     spr(o);
     int mid = t[o].l + t[o].r >> ;
     if(x <= mid) pl(o*,x,y,k);//如果x<=mid，说明左半区间包含要求区间
     if(y > mid) pl(o*+,x,y,k); //如果x>=mid ,说明右半区间包含要求区间
     t[o].tql = t[o*].tql + t[o*+].tql;
 }
 long long pr(int p,int x,int y){
     if(x <= t[p].l && y >= t[p].r) return t[p].tql;
     spr(p);
     int mid = t[p].l + t[p].r >> ;
     long long ans = ;
     if(x<=mid) ans += pr(p*,x,y);//如果x<=mid，说明左半子树包含答案
     if(y>mid) ans += pr(p*+,x,y);//如果x>mid,说明右半子树包含答案
     return ans;
 }
 int main(){
     cin >> n >> m;
     for(int i = ;i <= n; i++) cin >> a[i];
     build(,,n);//建树
     for(int i = ;i <= m; i++) {
         cin >> p;
         if(p == ) {
             cin >> x >> y >> pp;
             pl(,x, y, pp);
         }
         else {
             cin >> x >> y;
             cout << pr(,x,y) << endl;
         }
     }

     return ;
 }