Cod sursa(job #3124079)

Utilizator Robert_MitriRobert Mitri Robert_Mitri Data 26 aprilie 2023 20:18:13
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.39 kb
#include <fstream>
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");


int n,m;


int v[15005];
int aint[60006];

bool contains(int st,int dr,int x,int y)
{
    return x>=st && y<=dr;
}
int left_child(int val)
{
    return val*2;
}
int right_child(int val)
{
    return val*2+1;
}



void update(int nod,int st,int dr,int poz,int val)
{
    if(st==dr)
    {
        aint[nod]+=val;
        return;
    }
    int mid = (st+dr)/2;
    if(poz<=mid)
        update(left_child(nod),st,mid,poz,val);
    else
        update(right_child(nod),mid+1,dr,poz,val);
    aint[nod] = aint[left_child(nod)] + aint[right_child(nod)];
}
int sol=0;
void query(int nod,int st,int dr,int qst,int qdr)
{
    if(contains(qst,qdr,st,dr))
    {
        sol+=aint[nod];
        return;
    }
    int mid = (st+dr)/2;
    if(qst<=mid)
        query(left_child(nod),st,mid,qst,qdr);
    if(mid+1<=qdr)
        query(right_child(nod),mid+1,dr,qst,qdr);
}


int main()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        fin>>v[i];
        update(1,1,n,i,v[i]);
    }
    for(int i=1;i<=m;i++)
    {
        int t,a,b;
        fin>>t>>a>>b;
        if(t==0)
        {
            update(1,1,n,a,-b);
        }
        else
        {
            sol=0;
            query(1,1,n,a,b);
            fout<<sol<<'\n';
        }
    }

}