Cod sursa(job #2799006)

Utilizator alexioana_2006Apostolache Alexia alexioana_2006 Data 12 noiembrie 2021 10:26:38
Problema Datorii Scor 60
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.02 kb
#include <bits/stdc++.h>
using namespace std;
int arb[60005],n,m;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
void update(int nod, int a, int b, int poz, int val)
{
    if(a==b)
    {
        arb[nod]+=val;
        return ;
    }
    int mij=(a+b)>>1;
    if(poz<=mij)update((nod<<1),a,mij,poz,val);
    else update((nod<<1)+1,mij+1,b,poz,val);
    arb[nod]=arb[(nod<<1)]+arb[(nod<<1)+1];
}
int query(int nod, int a, int b, int qa, int qb)
{
    if(qa<=a && b<=qb)
        return arb[nod];
    int ans1,ans2,mij;
    ans1=ans2=0;
    mij=(a+b)>>1;
    if(qa<=mij)ans1=query((nod<<1),a,mij,qa,qb);
    if(qb>mij)ans2=query((nod<<1)+1,mij+1,b,qa,qb);
    return ans1+ans2;

}
int main()
{
    fin>>n>>m;
    for(int i=1; i<=n; ++i)
    {
        int x;
        fin>>x;
        update(1,1,n,i,x);
    }
    for(int i=1; i<=m; ++i)
    {
        int nr,a,b;
        fin>>nr>>a>>b;
        if(nr==0)update(1,1,n,a,-b);
        else fout<<query(1,1,n,a,b)<<'\n';
    }
    return 0;
}