Cod sursa(job #1684604)

Utilizator AdrianGotcaAdrian Gotca AdrianGotca Data 11 aprilie 2016 10:34:45
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.24 kb
#include <fstream>

using namespace std;

void update(int,int,int);
void query(int,int,int);
int n,m,val,poz,zi,start,stop,s,arb[400005];
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",&val);
        poz=i;
        update(1,n,1);
    }
    for (int i=1;i<=m;i++){
        int op;
        scanf("%d",&op);
        if (!op){
            scanf("%d%d",&zi,&val);
            poz=zi;
            val=-val;
            update(1,n,1);
        }
        else {
            scanf("%d%d",&start,&stop);
            s=0;
            query(1,n,1);
            printf("%d\n",s);
        }
    }
}

void update(int st,int dr,int n){
    if (st==dr){
        arb[n]+=val;
        return;
    }
    int mij=(st+dr)/2;
    if (poz<=mij){
        update(st,mij,2*n);
    }
    else {
        update(mij+1,dr,2*n+1);
    }
    arb[n]=arb[2*n]+arb[2*n+1];
    return;
}

void query(int st,int dr,int n){
    if (start<=st&&dr<=stop){
        s+=arb[n];
        return;
    }
    int mij=(st+dr)/2;
    if (start<=mij){
        query(st,mij,2*n);
    }
    if (mij<stop){
        query(mij+1,dr,2*n+1);
    }
}