codevs 1082 线段树练习 3（区间修改+区间求和）

题目描述 Description
给你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 <cstdio>  
#include <algorithm>    
#define lson l , m , rt << 1  
#define rson m + 1 , r , rt << 1 | 1  
#define LL long long  
using namespace std;
const int maxn =200005;  
LL lazy[maxn<<2];  
LL sum[maxn<<2]; 
int A[maxn]; 

void PushUp(int rt) 
{  
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];  
}  
void PushDown(int rt,int m)
{  
    if(lazy[rt])
    {  
        lazy[rt<<1]+=lazy[rt];  
        lazy[rt<<1|1]+=lazy[rt];  
        sum[rt<<1]+=lazy[rt]*(m-(m>>1));  
        sum[rt<<1|1]+=lazy[rt]*(m>>1);  
        lazy[rt]=0;  
    }  
}  

void build(int l,int r,int rt) 
{  
    lazy[rt] = 0;  
    if(l==r) 
    {  
        //scanf("%lld",&sum[rt]); 
        sum[rt]=A[l]; 
        return ;  
    }  
    int m=(l+r)>>1;  
    build(lson);  
    build(rson);  
    PushUp(rt);  
}  

void update(int L,int R,int c,int l,int r,int rt) 
{  
    if (L<=l&&r<=R) 
    {  
        lazy[rt]+=c;

        sum[rt]+=(LL)c*(r-l+1);  

        return ;  
    }  
    PushDown(rt,r-l+1);  
    int m=(l+r)>>1;  
    if (L<=m) update(L,R,c,lson);  
    if (m<R) update(L,R,c,rson);  
    PushUp(rt);  
}  

LL query(int L,int R,int l,int r,int rt) 
{  
    if (L<=l&&r<=R) 
    {  
        return sum[rt];  
    }  
    PushDown(rt,r-l+1);  

    int m=(l+r)>>1;  
    LL ret = 0;  
    if (L<=m) ret+=query(L,R,lson);  
    if (m<R) ret+=query(L,R,rson);  
    return ret;  
}  
int main() 
{  
    int n,q;  

    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&A[i]);

    build(1,n,1);  

    scanf("%d",&q);
    while(q--) 
    {  
        int f;
        scanf("%d",&f);
        if(f==1)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            update(x,y,z,1,n,1); 
        }
        else 
        {
            int x,y;
            scanf("%d%d",&x,&y);
            printf("%lld\n",query(x,y,1,n,1));
        }
    }  
    return 0;  
}