Cod sursa(job #2798996)

Utilizator alexioana_2006Apostolache Alexia alexioana_2006 Data 12 noiembrie 2021 10:18:30
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.91 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
int arb[400005],a,b,nr,n,m,i,x;
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 && qb>=b) return arb[nod];
    int mij=(a+b)>>1;
    int ans1=-1,ans2=-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(i=1;i<=n;i++)
    {fin>>x;
     update(1,1,n,i,x);
    }
 for(i=1;i<=m;i++)
 {
     fin>>nr>>a>>b;
     if(nr==0)
        update(1,1,n,a,-b);
     else fout<<query(1,1,n,a,b)<<'\n';
 }

    return 0;
}