P3372 【模板】线段树 1

题目描述
如题,已知一个数列,你需要进行下面两种操作:

将某区间每一个数加上 kk。
求出某区间每一个数的和。
输入格式
第一行包含两个整数 n, mn,m,分别表示该数列数字的个数和操作的总个数。

第二行包含 nn 个用空格分隔的整数,其中第 ii 个数字表示数列第 ii 项的初始值。

接下来 mm 行每行包含 33 或 44 个整数,表示一个操作,具体如下:

1 x y k:将区间 [x, y] 内每个数加上 kk。
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

#include<iostream>
#include<cstring>
#include<algorithm>
#include<string>
#include<cstdio>
using namespace std;
typedef long long ll;
const int maxn=100005;
ll a[maxn<<2],tag[maxn<<2],sum[maxn<<2];
ll n,m;
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
void pushup(ll rt){
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void move(ll rt,ll l, ll r,ll t){
    sum[rt]+=(r-l+1)*t;
    tag[rt]+=t;
}
void pushdown(ll rt,ll l,ll r){
	if(!tag[rt])return ;
    ll mid=(l+r)>>1;
    move(lson,tag[rt]);
    move(rson,tag[rt]);
    tag[rt]=0;
}
void build(ll rt,ll l,ll r){
    if(l==r){
        sum[rt]=a[l];
        return ;
    }
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}
void update(ll ql,ll qr,ll rt,ll l,ll r,ll num){
    if(ql<=l&&r<=qr){
        sum[rt]+=num*(r-l+1);
        tag[rt]+=num;
        return ;
    }
    pushdown(rt,l,r);
   ll mid=(l+r)>>1;
    if(ql<=mid){
        update(ql,qr,lson,num);
    }
    if(mid<qr){
        update(ql,qr,rson,num);
    }
    pushup(rt);
}
ll query(ll ql,ll qr,ll rt,ll l,ll r){
    if(ql<=l&&r<=qr){
        return sum[rt];
    }
   
    ll mid=(l+r)>>1;
	ll res=0;
	 pushdown(rt,l,r);
    if(ql<=mid){
        res+=query(ql,qr,lson);
    }
    if(mid<qr){
        res+=query(ql,qr,rson);
    }
    return res;

}
int main(){
   
   scanf("%lld%lld",&n,&m);
   for(int i=1;i<=n;i++){
   	scanf("%lld",&a[i]);
   }
   build(1,1,n);
   while(m--){
       int p;
       scanf("%d",&p);
       if(p==1){
           ll x,y,z;
           scanf("%lld%lld%lld",&x,&y,&z);
           update(x,y,1,1,n,z);
       }
       else {
           ll x,y;
           scanf("%lld%lld",&x,&y);
           printf("%lld\n",query(x,y,1,1,n));
       }
   }
   return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值