Cod sursa(job #1628416)

Utilizator duesakBourceanu Cristian duesak Data 4 martie 2016 00:13:48
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.32 kb
#include <fstream>
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
int varb[400201],n,m,ma;
void inserare(int nod, int st, int dr, int val, int pz){
    if(pz==st&&pz==dr){
            varb[nod]=val;
            return;
    }
    int mij=(st+dr)/2;
    if(mij>=pz)inserare(2*nod,st,mij,val,pz);
    else inserare(2*nod+1,mij+1,dr,val,pz);
    varb[nod]=varb[2*nod]+varb[2*nod+1];
}
void scadere(int nod, int st, int dr, int val, int pz){
    if(pz==st&&pz==dr){
            varb[nod]-=val;
            return;
    }
    int mij=(st+dr)/2;
    if(mij>=pz)scadere(2*nod,st,mij,val,pz);
    else scadere(2*nod+1,mij+1,dr,val,pz);
    varb[nod]=varb[2*nod]+varb[2*nod+1];
}
void cautare(int nod, int st, int dr, int a, int b){
    if(st>=a&&dr<=b){
        ma+=varb[nod];
        return;
    }
    int mij=(st+dr)/2;
    if(a<=mij)cautare(2*nod,st,mij,a,b);
    if(mij<b)cautare(2*nod+1,mij+1,dr,a,b);
}
int main(){
    int a,b,caz;
    fin>>n>>m;
    for(int i=1;i<=n;i++){
        fin>>caz;
        inserare(1,1,n,caz,i);
    }
    for(int i=1;i<=m;i++){
        fin>>caz>>a>>b;
        if(caz){
             ma=0;
             cautare(1,1,n,a,b);
             fout<<ma<<'\n';
        }
        else{
            scadere(1,1,n,b,a);
        }
    }
    return 0;
}