Cod sursa(job #2859351)

Utilizator toma_ariciuAriciu Toma toma_ariciu Data 1 martie 2022 10:56:58
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.52 kb
#include <iostream>
#include <fstream>

using namespace std;

#pragma GCC optimize ("Ofast")

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

int n, q;
int v[15005], arbint[60005];

void init_arbint(int nod, int st, int dr)
{
    if(st == dr)
    {
        arbint[nod] = v[st];
        return;
    }
    int med = (st + dr) / 2;
    init_arbint(2 * nod, st, med);
    init_arbint(2 * nod + 1, med + 1, dr);
    arbint[nod] = arbint[2 * nod] + arbint[2 * nod + 1];
}

void update(int nod, int st, int dr, int poz, int val)
{
    if(st == dr)
    {
        arbint[nod] -= val;
        return;
    }
    int med = (st + dr) / 2;
    if(poz <= med)
        update(2 * nod, st, med, poz, val);
    else
        update(2 * nod + 1, med + 1, dr, poz, val);
    arbint[nod] = arbint[2 * nod] + arbint[2 * nod + 1];
}

int query(int nod, int st, int dr, int qst, int qdr)
{
    if(qst <= st && dr <= qdr)
        return arbint[nod];
    int med = (st + dr) / 2, sum = 0;
    if(qst <= med)
        sum += query(2 * nod, st, med, qst, qdr);
    if(med + 1 <= qdr)
        sum += query(2 * nod + 1, med + 1, dr, qst, qdr);
    return sum;
}

int main()
{
    fin >> n >> q;
    for(int i = 1; i <= n; i++)
        fin >> v[i];
    init_arbint(1, 1, n);
    for(int i = 1; i <= q; i++)
    {
        int c, x, y;
        fin >> c >> x >> y;
        if(c == 0)
            update(1, 1, n, x, y);
        if(c == 1)
            fout << query(1, 1, n, x, y) << '\n';
    }
    return 0;
}