Cod sursa(job #2918156)

Utilizator Theo14Ancuta Theodor Theo14 Data 10 august 2022 11:15:51
Problema Datorii Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.1 kb
#include<bits/stdc++.h>
using namespace std;

ifstream f("datorii.in");
ofstream g("datorii.out");

int v[400004];

void update(int nod, int st, int dr, int poz, int val)
{
    if(st==dr)
    {
        v[nod]+=val;
        return;
    }
    int mij=(st+dr)/2;
    if(poz<=mij)
        update(2*nod,st,mij,poz,val);
    else
        update(2*nod+1,mij+1,dr,poz,val);
    v[nod]=v[2*nod]+v[2*nod+1];
}

int query(int nod, int st, int dr, int l, int r)
{
    if(r<st || l>dr)
        return 0;
    if(l<=st && r>=dr)
        return v[nod];
    int mij=(st+dr)/2;
    if(dr<=mij)
        return query(2*nod,st,mij,l,r);
    else
    if(st>=mij+1)
        return query(2*nod+1,mij+1,dr,l,r);
    else
        return query(2*nod,st,mij,l,r)+query(2*nod+1,mij+1,dr,l,r);
}

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