Cod sursa(job #2625304)

Utilizator albertyoAlbert Mindrescu albertyo Data 5 iunie 2020 21:06:20
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.5 kb
#include <fstream>
#define N 30101
using namespace std;

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

int n, m, a[N], s[N];

void Creare(int nod,int st,int dr) {

    int mij;
    mij = st + (dr - st) / 2;
    if(st == dr) s[nod] = a[st];
    else {

        Creare(nod * 2, st, mij);
        Creare(nod * 2 + 1, mij + 1, dr);
        s[nod] = s[nod * 2] + s[nod * 2 + 1];
    }
}

int Suma(int nod, int st, int dr, int pozs, int pozd) {

    int mij;
    if(st >= pozs && dr <= pozd) return s[nod];
    else if(st > pozd) return 0;
    else if(dr < pozs) return 0;
    else {
        mij = st + (dr - st) / 2;
        return Suma(nod * 2, st, mij, pozs, pozd) + Suma(nod * 2 + 1, mij + 1, dr, pozs, pozd);
    }
}

int Cautare(int nod,int st,int dr,int x) {

    int mij;
    if(st == dr && st == x) return nod;
    else {
        mij = st + (dr - st) / 2;
        if(mij < x) return Cautare(nod * 2 + 1, mij + 1, dr, x);
        else return Cautare(nod * 2, st, mij, x);
    }
}

void Actualizare(int poz,int y) {

    s[poz] -= y;
    if(poz != 1) Actualizare(poz / 2, y);
}

int main()
{
    int i, op, x, y, poz;

    fin >> n >> m;

    for(i = 1; i <= n; i++)
        fin >> a[i];

    Creare(1,1,n);
    for(i = 1; i <= m; i++) {

        fin >> op >> x >> y;
        if(op == 1) fout << Suma(1, 1, n, x, y) << "\n";
        if(op == 0) {

            poz = Cautare(1, 1, n, x);
            Actualizare(poz, y);
        }
    }

    return 0;
}