洛谷 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[10000005],p,x,y,pp;
struct kkk{
    int l,r;//l、r分别代表此节点所含信息区间左端和右端 
    long long add,tql;//add是lazy标记，tql代表此节点所含区间和。 
}t[10000005];
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 >> 1;
    build(2*q, w, mid);     //建左子树 
    build(2*q+1, mid+1, e); //建右子树 
    t[q].tql = t[q*2].tql + t[q*2+1].tql; //一个节点的信息由它的左右儿子得来 
}
void spr(int o) {//运用lazy标记进行区间修改 
    if(t[o].add) {
        t[2*o].tql += t[o].add * (t[2*o].r - t[2*o].l + 1);
        t[o*2+1].tql += t[o].add * (t[o*2+1].r - t[o*2+1].l + 1);
        t[o*2].add += t[o].add;
        t[o*2+1].add += t[o].add;
        t[o].add = 0;
    }
}
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 + 1);
        t[o].add += k;
        return ;
    }
    spr(o);
    int mid = t[o].l + t[o].r >> 1;
    if(x <= mid) pl(o*2,x,y,k);//如果x<=mid，说明左半区间包含要求区间 
    if(y > mid) pl(o*2+1,x,y,k); //如果x>=mid ,说明右半区间包含要求区间 
    t[o].tql = t[o*2].tql + t[o*2+1].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 >> 1;
    long long ans = 0;
    if(x<=mid) ans += pr(p*2,x,y);//如果x<=mid，说明左半子树包含答案 
    if(y>mid) ans += pr(p*2+1,x,y);//如果x>mid,说明右半子树包含答案 
    return ans;
}
int main(){
    cin >> n >> m;
    for(int i = 1;i <= n; i++) cin >> a[i];
    build(1,1,n);//建树 
    for(int i = 1;i <= m; i++) {
        cin >> p;
        if(p == 1) {
            cin >> x >> y >> pp;
            pl(1,x, y, pp);
        }
        else {
            cin >> x >> y;
            cout << pr(1,x,y) << endl;
        }
    }
    
    return 0;
}