Cod sursa(job #2866561)

Utilizator andrei_laurentiuRadu Andrei-Laurentiu andrei_laurentiu Data 9 martie 2022 19:53:02
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.55 kb
#include <iostream>
#include <fstream>

using namespace std;

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

int v[15001];
long long arbore[4 * 15000 + 66];

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

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

        arbore[nod] = arbore[nod * 2] + arbore[nod * 2 + 1];
    }
}

long long query(int nod, int st, int dr, int a, int b)
{
    if(a <= st && dr <= b)
        return arbore[nod];
    else
    {
        int mij = (st + dr) / 2, ans1 = 0, ans2 = 0;
        if(a <= mij)
            ans1 = query(nod * 2, st, mij, a, b);
        if(mij < b)
            ans2 = query(nod * 2 + 1, mij + 1, dr, a, b);
        return ans1 + ans2;
    }
}

int main()
{
    int n, m, a, b, c;

    fin >> n >> m;
    for(int i = 1; i <= n; ++i)
        fin >> v[i];

    build(1, 1, n);


    for(int i = 1; i <= m; ++i)
    {
        fin >> c >> a >> b;

        if(c == 1)
            fout << query(1, 1, n, a, b)<< '\n';
        else
            update(1, 1, n, a, b);
    }
    return 0;
}