Cod sursa(job #3153686)

Utilizator Mihai_PopescuMihai Popescu Mihai_Popescu Data 30 septembrie 2023 18:08:52
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.63 kb
#include <fstream>
using namespace std;

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

#define NMAX 4 * 15005

int v[NMAX];

void build(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) {
        build(2 * nod, st, mij, poz, val);
    } else {
        build(2 * nod + 1, mij + 1, dr, poz, val);
    }
    v[nod] += val;
}

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] -= val;
}

void query(int nod, int st, int dr, int p, int q, int &sum) {
    if (p <= st && dr <= q) {
        sum += v[nod];
        return ;
    }
    int mij = (st + dr) / 2;
    if (p <= mij) {
        query(2 * nod, st, mij, p, q, sum);
    }
    if (mij < q) {
        query(2 * nod + 1, mij + 1, dr, p, q, sum);
    }
}

int main() {
    int n, m;
    fin >> n >> m;
    for (int i = 1; i <= n; ++i) {
        int a;
        fin >> a;

        build(1, 1, n, i, a);
    }

    while (m--) {
        bool cod;
        fin >> cod;
        if (cod == 0) {
            int t, v;
            fin >> t >> v;

            update(1, 1, n, t, v);
        } else {
            int p, q;
            fin >> p >> q;

            int sum = 0;
            query(1, 1, n, p, q, sum);
            fout << sum << '\n';
        }
    }
    return 0;
}