WHYZOJ-#53 线段树区间修改(线段树)

【题目描述】：

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

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的结果。

【样例输入】：

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

【样例输出】：

11
8
20

【时间限制、数据范围及描述】：

时间：1s 空间：128M

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

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

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

（数据保证在int64/long long数据范围内）

 

 感觉通过加和乘那一题，我的线段树水平有了质的提升，还是那个原则，打延迟标记的时候当前节点也要更新，并且，只要需要访问子节点，就一定要标记下传，无论是查询还是更新！！！

 

#include "bits/stdc++.h"
#define mem(a,b) memset(a,b,sizeof(a))
#define lson rt<<1,l,m
#define rson rt<<1|1,m+1,r
using namespace std;
typedef long long LL;
const int MAX=100005;
int n,m;
LL sum[MAX*4],la[MAX*4];
void PushUp(int rt){
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void PushDown(int rt,int l,int r){
    int m=(l+r)>>1;
    if (la[rt]!=0){
        la[rt<<1]+=la[rt];la[rt<<1|1]+=la[rt];
        sum[rt<<1]+=la[rt]*(m-l+1);sum[rt<<1|1]+=la[rt]*(r-m);
        la[rt]=0;
    }
}
void build(int rt,int l,int r){
    if (l==r){
        scanf("%lld",&sum[rt]);
        la[rt]=0;
        return;
    }la[rt]=0;
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    PushUp(rt);
}
void update(int rt,int l,int r,LL x,LL y,LL z){
    if (x<=l && r<=y){
        la[rt]+=z;
        sum[rt]+=(r-l+1)*z;
        return;
    }
    int m=(l+r)>>1;
    PushDown(rt,l,r);
    if (x<=m)
        update(lson,x,y,z);
    if (y>m)
        update(rson,x,y,z);
    PushUp(rt);
}
LL search(int rt,int l,int r,int x,int y){
    if (x<=l && r<=y) return sum[rt];
    LL an=0;
    int m=(l+r)>>1;
    PushDown(rt,l,r);
    if (x<=m)
        an+=search(lson,x,y);
    if (y>m)
        an+=search(rson,x,y);
    return an;
}
int main(){
    freopen ("segtree.in","r",stdin);
    freopen ("segtree.out","w",stdout);
    int i,j;
    LL x,y,z,w;
    scanf("%d%d",&n,&m);
    build(1,1,n);
    for (i=1;i<=m;i++){
        scanf("%lld",&w);
        if (w==1){
            scanf("%lld%lld%lld",&x,&y,&z);
            update(1,1,n,x,y,z);
        }
        if (w==2){
            scanf("%lld%lld",&x,&y);
            printf("%lld
",search(1,1,n,x,y));
        }
    }
    return 0;
}