洛谷 P3368 【模板】树状数组 2

题目：树状数组 2

思路：线段树模板题

代码：

#include<bits/stdc++.h>
using namespace std;

#define maxn 500000
#define ll long long

struct Node{
    ll x,lzy;
};

ll n,m;
ll c[maxn+5];
Node a[maxn*10+5];

ll P,Q;

void push_down(ll o) {
    a[o*2].lzy+=a[o].lzy,a[o*2+1].lzy+=a[o].lzy;
    a[o*2].x+=a[o].lzy,a[o*2+1].x+=a[o].lzy;
    a[o].lzy=0;
}

void make_tree(ll o,ll L,ll R) {
    if(L==R) {
        a[o].x=c[L];
        return ;
    }
    ll mid=L+(R-L)/2;
    ll lch=o*2,rch=o*2+1;
    make_tree(lch,L,mid),make_tree(rch,mid+1,R);
}

void Add(ll o,ll L,ll R,ll k){
    push_down(o);
    if(P<=L&&R<=Q){
        a[o].lzy+=k;
        a[o].x+=k;
        return ;
    }
    if(P>R||Q<L) return ;
    ll mid=L+(R-L)/2;
    ll lch=o*2,rch=o*2+1;
    Add(lch,L,mid,k),Add(rch,mid+1,R,k);
}

ll Query(ll o,ll L,ll R,ll k){
    push_down(o);
    if(L==R&&k==L){
        return a[o].x;
    }
    if(k>R||k<L) return -1;
    ll mid=L+(R-L)/2;
    ll lch=o*2,rch=o*2+1;
    return max(Query(lch,L,mid,k),Query(rch,mid+1,R,k));
}

int main() {
    scanf("%lld%lld",&n,&m);
    for(ll i=1;i<=n;i++) scanf("%lld",&c[i]);
    make_tree(1,1,n);
    while(m--) {
        ll opr;
        scanf("%lld",&opr);
        if(opr==1) {
            ll k;
            scanf("%lld%lld%lld",&P,&Q,&k);
            Add(1,1,n,k);
        } else {
            ll k;
            scanf("%lld",&k);
            ll ans=Query(1,1,n,k);
            printf("%lld\n",ans);
        }
    }
    return 0;
}