Cod sursa(job #2084533)

Utilizator andrei2000mAndrei Moldoveanu andrei2000m Data 9 decembrie 2017 10:35:51
Problema Datorii Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.1 kb
#include <fstream>

using namespace std;
ifstream f("datorii.in");
ofstream g("datorii.out");
int n,m,arb[60001];
void citire(int nod,int a,int b,int poz,int val)
{
    int mij;
    arb[nod]+=val;
    if(a!=b){mij=(a+b)/2;
            if(poz<=mij) citire(2*nod,a,mij,poz,val);
            else citire(2*nod+1,mij+1,b,poz,val);
            }
}
void update(int nod,int a,int b,int poz,int val)
{
    int mij;
    arb[nod]-=val;
    if(a!=b) {  mij=(a+b)/2;
            if(poz<=mij) update(2*nod,a,mij,poz,val);
            else update(2*nod+1,mij+1,b,poz,val);
         }
}
int query(int nod,int a,int b,int qa,int qb)
{
    int mij,rez1=0,rez2=0;
    if(qa<=a&&b<=qb) return arb[nod];
    mij=(a+b)/2;
    if(qa<=mij) rez1=query(2*nod,a,mij,qa,qb);
    if(mij<qb) rez2=query(2*nod+1,mij+1,b,qa,qb);
    return rez1+rez2;
}
int main()
{   int i,c,a,b,x;
    f>>n>>m;
    for(i=1;i<=n;i++)
    {
        f>>x;
        citire(1,1,n,i,x);
    }
    for(i=1;i<=m;i++)
    {   f>>c>>a>>b;
        if(c==0) update(1,1,n,a,b);
        else g<<query(1,1,n,a,b)<<endl;
    }
    return 0;
}