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;  
}  
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值