Cod sursa(job #3152979)

Utilizator tudor_costinCostin Tudor tudor_costin Data 27 septembrie 2023 13:23:33
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.58 kb
#include <fstream>

using namespace std;
ifstream cin("datorii.in");
ofstream cout("datorii.out");
const int Nmax=15005;
int arb[Nmax],v[Nmax];
void build(int st, int dr, int nod)
{
    if(st==dr)
    {
        arb[nod]=v[st];
    }
    else
    {
        int mij=(st+dr)/2;
        build(st,mij,2*nod);
        build(mij+1,dr,2*nod+1);
        arb[nod]=arb[2*nod]+arb[2*nod+1];
    }
    return;
}
void update(int st,int dr,int poz,int val,int nod)
{
    if(st==dr)
    {
        arb[nod]-=val;
    }
    else
    {
        int mij=(st+dr)/2;
        if(poz<=mij)
        {
            update(st,mij,poz,val,2*nod);
        }
        else
        {
            update(mij+1,dr,poz,val,2*nod+1);
        }
        arb[nod]=arb[2*nod]+arb[2*nod+1];
    }
}
int query(int st,int dr,int qst,int qdr,int nod)
{
    if(qst<=st && qdr>=dr)
    {
        return arb[nod];
    }
    else
    {
        int mij=(st+dr)/2;
        if(qdr<=mij)
        {
            return query(st,mij,qst,qdr,2*nod);
        }
        if(qst>mij)
        {
            return query(mij+1,dr,qst,qdr,2*nod+1);
        }
        return query(st,mij,qst,qdr,2*nod)+query(mij+1,dr,qst,qdr,2*nod+1);
    }
}
int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        cin>>v[i];
    }
    build(1,n,1);
    int a,b,c;
    for(int i=1;i<=m;i++)
    {
        cin>>c>>a>>b;
        if(c==0)
        {
            update(1,n,a,b,1);
        }
        if(c==1)
        {
            cout<<query(1,n,a,b,1)<<'\n';
        }
    }
    return 0;
}