Cod sursa(job #2087923)

Utilizator andrei2000mAndrei Moldoveanu andrei2000m Data 14 decembrie 2017 15:53:22
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.13 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 querry(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=querry(2*nod,a,mij,qa,qb);
    if(mij<qb) rez2=querry(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<<querry(1,1,n,a,b)<<"\n";
    }
    return 0;
}