Cod sursa(job #3128935)

Utilizator solicasolica solica solica Data 11 mai 2023 16:40:47
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.65 kb
#include <fstream>

using namespace std;

#define NMAX 15008

ifstream fin ("datorii.in");
ofstream fout ("datorii.out");

int n,q,i,j,tip,stanga,dreapta;
int a[NMAX];
int tree[4*NMAX];

void build (int node, int st, int dr)
{
    if (st==dr)
    {
        tree[node]=a[st];
    }
    else
    {
        int mij=(st+dr)/2;
        build(node*2,st,mij);
        build(node*2+1,mij+1,dr);
        tree[node]=tree[node*2]+tree[node*2+1];
    }
}

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

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

int main()
{
    fin>>n>>q;
    for (i=1; i<=n; i++)
    {
        fin>>a[i];
    }
    build(1,1,n);
    for (i=1; i<=q; i++)
    {
        fin>>tip>>stanga>>dreapta;
        if (tip==0)
        {
            update(1,1,n,stanga,dreapta);
        }
        if (tip==1)
        {
            fout<<query(1,1,n,stanga,dreapta)<<'\n';
        }
    }
    return 0;
}