Cod sursa(job #3152049)

Utilizator ungureanubogdan07Ungureanu Bogdan ungureanubogdan07 Data 23 septembrie 2023 17:20:27
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.49 kb
#include <fstream>

using namespace std;

ifstream cin("date.in");
ofstream cout("date.out");

int v[150001], aint[600001];
int n, q;

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

void update(int nod, int st, int dr, int poz, int val)
{
    if(st == dr) {
        aint[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);
    }
    aint[nod] = aint[2 * nod] + aint[2 * nod + 1];
}

int query(int nod, int st, int dr, int stq, int drq)
{
    if(st >= stq && dr <= drq) {
        return aint[nod];
    }   
    int mij = (st + dr) / 2;
    if(mij >= drq) {
        return query(2 * nod, st, mij, stq, drq);
    }
    if(mij + 1 <= stq) {
        return query(2 * nod + 1, mij + 1, dr, stq, drq);
    }
    return query(2 * nod, st, mij, stq, drq) + query(2 * nod + 1, mij + 1, dr, stq, drq);
}

int main()
{
    cin >> n >> q;
    for(int i = 1; i <= n; ++i) cin >> v[i];
    build(1, 1, n);
    while(q--) {
        int op, x, y;
        cin >> op >> x >> y;
        if(op == 0) {
            update(1, 1, n, x, y);
        }
        else {
            cout << query(1, 1, n, x, y) << '\n';
        }
    }
}