Cod sursa(job #3236428)

Utilizator stefanrotaruRotaru Stefan-Florin stefanrotaru Data 28 iunie 2024 16:12:24
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.31 kb
#include <fstream>

using namespace std;

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

int n, m, val, caz, a, b, AI[45015], sum;

void update(int nod, int st, int dr, int poz, int val)
{
    if (st == dr) {
        AI[nod] += val;
        return;
    }
    else {
        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);
        }
    }

    AI[nod] = AI[2 * nod] + AI[2 * nod | 1];
}

void query(int nod, int st, int dr, int &sum, int a, int b)
{
    if (a <= st && dr <= b) {
        sum += AI[nod];
        return;
    }
    else {
        int mij = (st + dr) / 2;

        if (a <= mij) {
            query(2 * nod, st, mij, sum, a, b);
        }

        if (b > mij) {
            query(2 * nod | 1, mij + 1, dr, sum, a, b);
        }
    }
}

int main()
{
    f >> n >> m;

    for (int i = 1; i <= n; ++i) {
        f >> val;
        update(1, 1, n, i, val);
    }

    while (m--) {
        f >> caz >> a >> b;

        if (caz == 0) {
            update(1, 1, n, a, -b);
        }
        else {
            sum = 0;

            query(1, 1, n, sum, a, b);

            g << sum << '\n';
        }
    }

    return 0;
}