Cod sursa(job #2757000)

Utilizator tryharderulbrebenel mihnea stefan tryharderul Data 3 iunie 2021 15:37:05
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.26 kb
#include <bits/stdc++.h>
#define NMAX 45003

using namespace std;

int n,m,x,poz,ans;
int type,a,b;
int t[NMAX];
int v[15003];

void build(int nod,int st,int dr){
    if(st==dr){
        t[nod]+=v[st];
        return;
    }
    int mid=(st+dr)/2;
    build(nod*2,st,mid);
    build(nod*2+1,mid+1,dr);
    t[nod]=t[nod*2]+t[nod*2+1];
}
void update(int nod,int st,int dr){
    if(st==dr){
        t[nod]-=b;
        return;
    }
    int mid=(st+dr)/2;
    if(poz<=mid){
        update(nod*2,st,mid);
    }else{
        update(nod*2+1,mid+1,dr);
    }
    t[nod]=t[nod*2]+t[nod*2+1];
}

void query(int nod,int st,int dr){
    if(a<=st && dr<=b){
        ans+=t[nod];
        return;
    }
    int mid=(st+dr)/2;
    if(a<=mid)query(nod*2,st,mid);
    if(b>mid)query(nod*2+1,mid+1,dr);
}
int main()
{
    freopen("datorii.in","r",stdin);
    freopen("datorii.out","w",stdout);
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&v[i]);
    }
    build(1,1,n);
    while(m--){
        scanf("%d %d %d",&type,&a,&b);
        if(type==0){
            poz=a;
            update(1,1,n);
        }else{
            ans=0;
            query(1,1,n);
            printf("%d\n",ans);
        }
    }

    return 0;
}