Cod sursa(job #2579216)

Utilizator mihaimodiMihai Modi mihaimodi Data 12 martie 2020 11:32:50
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.12 kb
#include<fstream>
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
int aint[45000];
int n,m,t,p,q,poz,val,x,y;
void update(int nod, int st, int dr)
{
    if(st==dr)
    {
        aint[nod]+=val;
        return;
    }
    int mid=(st+dr)/2;
    if(poz<=mid)
        update(2*nod,st,mid);
    else
        update(2*nod+1,mid+1,dr);
    aint[nod]=aint[2*nod]+aint[2*nod+1];
}
int query(int nod, int st, int dr)
{
    if(x<=st&&dr<=y)
        return aint[nod];

    int mid=(st+dr)/2;
    if(x<=mid&&y>mid)
        return query(2*nod+1,mid+1,dr)+query(2*nod,st,mid);
    else if(x<=mid)
        return query(2*nod,st,mid);
    else if(y>mid)
        return query(2*nod+1,mid+1,dr);
}
int main()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        fin>>val;
        poz=i;
        update(1,1,n);
    }
    for(int i=1;i<=m;i++)
    {
        fin>>t>>p>>q;
        if(t==0)
        {
            val=-q,poz=p;
            update(1,1,n);
        }
        else
        {
            x=p,y=q;
            fout<<query(1,1,n)<<'\n';
        }
    }
    return 0;
}