Cod sursa(job #2904153)

Utilizator anamaria_panait.10Panait Ana-Maria anamaria_panait.10 Data 17 mai 2022 22:32:39
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.49 kb
#include <fstream>

using namespace std;

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

int tree[60020];

void build(int nod, int st, int dr, int x, int poz) {
    if(st == dr) {
        tree[nod] = x;
    } else {
        int mij = (st + dr) / 2;
        if(poz <= mij)
            build(nod * 2, st, mij, x, poz);
        else
            build(nod * 2 + 1, mij + 1, dr, x, poz);
        tree[nod] = tree[nod * 2] + tree[nod * 2 + 1];
    }
}

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

int sum(int nod, int st, int dr, int pos_st, int pos_dr) {
    if(pos_st > pos_dr)
        return 0;

    if(pos_st == st && pos_dr == dr)
        return tree[nod];

    int mij = (st + dr) / 2;
    return sum((nod * 2), st, mij, pos_st, min(pos_st, mij)) + sum((nod * 2) + 1, mij + 1, dr, max(pos_st, mij + 1), pos_dr);
}

int main() {
    int n, m, x, y, op;
    in >> n >> m;
    for(int i = 1; i <= n; i++) {
        in >> x;
        build(1, 1, n, x, i);
    }

    for(int i = 1; i <= m; i++){
        in >> op >> x >> y;
        if(op == 0)
            update(1, 1, n, x, y);
        else
            out << sum(1, 1, n, x, y) << '\n';
    }
    return 0;
}