线段树模板1 - 求区间和

Posted 2021-02-09 xxrlz

今天再次入门线段树

然而还是入门失败

push_up 和 push_down 不是太懂啊

线段树在有结合律性质的区间操作都可以用

然而我并不会

先开个坑吧

#include<bits/stdc++.h> 
#define MAXN  1000001
#define ll long long
#define ls(o)   (o<<1)
#define rs(o)   (o<<1|1) 
using namespace std;
ll n,m,a[MAXN],ans[MAXN<<2],tag[MAXN<<2];

void push_up(ll p)
{
    ans[p] = ans[ls(p)]+ans[rs(p)];
}
void build(ll p,ll l,ll r)
{
    tag[p] = 0;
    if(l==r){
        ans[p] = a[l];
        return;
    }
    ll mid = (l+r)>>1;
    build(ls(p),l,mid);
    build(rs(p),mid+1,r);
    push_up(p);
}

void push_down(ll p,ll l,ll r)
{
    ll mid =(l+r)>>1;
    tag[ls(p)] += tag[p];
    ans[ls(p)] += tag[p]*(mid-l+1);
    tag[rs(p)] += tag[p];
    ans[rs(p)] += tag[p]*(r-mid);
    tag[p] = 0;
}
void update(ll nl,ll nr,ll l,ll r,ll p,ll k)
{
    if(nl<=l && r<=nr)
    {
        ans[p] +=k*(r-l+1);
        tag[p] +=k;
        return ;
    }
    push_down(p,l,r);
    ll mid = (l+r)>>1;
    if(nl<=mid) update(nl,nr,l,mid,ls(p),k);
    if(nr>mid)  update(nl,nr,mid+1,r,rs(p),k);
    push_up(p);
}
ll query(ll q_x,ll q_y,ll l,ll r,ll p)
{
    ll res = 0;
    if(q_x<=l&&r<=q_y)  return ans[p];
    ll mid = (l+r)>>1;
    push_down(p,l,r);
    if(q_x<=mid)    res+=query(q_x,q_y,l,mid,ls(p));
    if(q_y>mid)     res+=query(q_x,q_y,mid+1,r,rs(p));
    return res;
}
int main(){
    ll a1,b,c,d,e,f;
    cin >> n >> m;
    for(ll i=1;i<=n;++i)
    {
        scanf("%lld",&a[i]);
    }
    build(1,1,n);
    while(m--)
    {
        scanf("%lld",&a1);
        if(a1==1){
            scanf("%lld%lld%lld",&b,&c,&d);
            update(b,c,1,n,1,d); 
        }
        else{
            scanf("%lld%lld",&e,&f);
            printf("%lld
",query(e,f,1,n,1));
        }
    }
    return 0;
}

事实证明半年过去我还是没学会QAQ