codevs 1082 线段树练习 3

1082 线段树练习 3 区间修改，区间查询；
时间限制: 3 s
空间限制: 128000 KB
题目等级 : 大师 Master
给你N个数，有两种操作：

1：给区间[a,b]的所有数增加X

2：询问区间[a,b]的数的和。

输入描述 Input Description
第一行一个正整数n，接下来n行n个整数，

再接下来一个正整数Q，每行表示操作的个数，

如果第一个数是1，后接3个正整数，

表示在区间[a,b]内每个数增加X,如果是2，

表示操作2询问区间[a,b]的和是多少。

pascal选手请不要使用readln读入

输出描述 Output Description
对于每个询问输出一行一个答案

样例输入 Sample Input
3

1

2

3

2

1 2 3 2

2 2 3

样例输出 Sample Output
9

数据范围及提示 Data Size & Hint
数据范围

1<=n<=200000

1<=q<=200000

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int sz=200000+50;
struct seg_tree{
    int l,r;
    long long sum;
    long long ax;
} t[sz<<2];
int n,a,q,x,y,z;
long long  v[sz];
void updata(int m){
    t[m].sum=t[m<<1].sum+t[m<<1|1].sum;
    return ;
}
void build(int m,int ll,int rr){
    t[m].l=ll,t[m].r=rr;
    if(ll==rr) {
        t[m].sum=v[ll];
        return ;
    }
    int mid=(ll+rr)>>1;
    build(m<<1,ll,mid);
    build(m<<1|1,mid+1,rr);
    updata(m);
}
void add(int m,int vv){
    t[m].sum+=(t[m].r-t[m].l+1)*vv;
    t[m].ax+=vv;
    return ;
}
void spread(int m){
    if(t[m].ax){
        add(m<<1,t[m].ax);
        add(m<<1|1,t[m].ax);
        t[m].ax=0;
        //这里要改为0！！ 否则用while会死循环
    }
    return ;
}
long long ask(int m,int ll,int rr){
    if(t[m].l>=ll&&t[m].r<=rr) return t[m].sum;
    int mid=t[m].l+t[m].r>>1;
    long long ans=0;
    spread(m);
    if(mid>=ll) ans+=ask(m<<1,ll,rr);
    if(mid<rr) ans+=ask(m<<1|1,ll,rr);
    return ans;
}
void change(int m,int ll,int rr,int vv){
    if(t[m].l>=ll&&t[m].r<=rr) {
        add(m,vv);return;
    }
    int mid=(t[m].l+t[m].r)>>1;
    spread(m);
    if(mid>=ll) change(m<<1,ll,rr,vv);
    if(mid<rr) change(m<<1|1,ll,rr,vv);
    updata(m);
    return ;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%lld",&v[i]);
    build(1,1,n);

    scanf("%d",&q);
    while(q--){
        scanf("%d",&a);
        if(a==1){
            scanf("%d%d%d",&x,&y,&z);
            change(1,x,y,z);
        //  for(int i=1;i<=n*4;i++)
        //printf("i= %d  t[i].l= %d  t[i].r= %d  t[i].sum= %d  t[i].ax= %d\n",i,t[i].l,t[i].r,t[i].sum,t[i].ax);
        }
        else scanf("%d%d",&x,&y),printf("%lld\n",ask(1,x,y));
    }
    return 0;
}