Cod sursa(job #3309050)

Utilizator CheeseEaterHackRoman Alex CheeseEaterHack Data 31 august 2025 15:13:17
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.42 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");

int arb[60003], v[15001], n, q, caz, a, b, p;

void build(int nod=1, int st=1, int dr=n)
{
    if (st==dr)
    {
        arb[nod]=v[st];
        return;
    }
    int mij=(st+dr)/2;
    build(nod*2, st, mij);
    build(2*nod+1, mij+1, dr);
    arb[nod]=arb[nod*2]+arb[2*nod+1];
}

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

int query(int a, int b, int nod=1, int st=1, int dr=n)
{
    if (a<=st and b>=dr)
    {
        return arb[nod];
    }
    int sum=0;
    int mij=(st+dr)/2;
    if (a<=mij)
    {
        sum+=query(a, b, 2*nod, st, mij);
    }
    if (mij<b)
    {
        sum+=query(a, b, 2*nod+1, mij+1, dr);
    }
    return sum;
}

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