【求助】线段树 0pts
  • 板块P2357 守墓人
  • 楼主Bei_Gua_
  • 当前回复4
  • 已保存回复4
  • 发布时间2024/1/2 19:04
  • 上次更新2024/1/2 23:03:05
查看原帖
【求助】线段树 0pts
889498
Bei_Gua_楼主2024/1/2 19:04
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=2e5+5;
int n,m;
int a[N],tree[N<<2],tag[N<<2];
inline int ls(int x){return x<<1;}
inline int rs(int x){return x<<1|1;}
inline void push_up(int p){
    tree[p]=tree[ls(p)]+tree[rs(p)];
}
inline void build(int p,int l,int r){
    if (l==r){tree[p]=a[l];return;}
    int mid=(l+r)>>1;
    build(ls(p),l,mid);
    build(rs(p),mid+1,r);
    push_up(p);
}
inline void f(int p,int k,int l,int r){
    tag[p]+=k;
    tree[p]+=k*(r-l+1);
}
inline void push_down(int p,int l,int r){
    int mid=(l+r)>>1;
    f(ls(p),tag[p],l,mid);
    f(rs(p),tag[p],mid+1,r);
    tag[p]=0;
}
inline void update(int nl,int nr,int l,int r,int p,int k){
    if (nl<=l&&r<=nr){
        tag[p]+=k;
        tree[p]+=k*(r-l+1);
        return;
    }
    push_down(p,l,r);
    int 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);
}
inline int query(int nx,int ny,int l,int r,int p){
    int res=0;
    if (nx<=l&&r<=ny){
        return tree[p];
    }
    int mid=(l+r)>>1;
    if (nx<=mid)res+=query(nx,ny,l,mid,ls(p));
    if (ny>mid)res+=query(nx,ny,mid+1,r,rs(p));
    return res;
}
signed main(){
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    cin>>n>>m;
    for (int i=1;i<=n;i++)cin>>a[i];
    build(1,1,n);
    for (int i=1;i<=m;i++){
        int op;
        cin>>op;
        if (op==1){
            int l,r,k;
            cin>>l>>r>>k;
            update(l,r,1,n,1,k);
        }else if (op==2){
            int k;
            cin>>k;
            update(1,1,1,n,1,k);
        }else if (op==3){
            int k;
            cin>>k;
            update(1,1,1,n,1,-k);
        }else if (op==4){
            int l,r;
            cin>>l>>r;
            cout<<query(l,r,1,n,1)<<"\n";
        }else if (op==5){
            cout<<query(1,1,1,n,1)<<"\n";
        }
    }
}
2024/1/2 19:04
加载中...