Cod sursa(job #2904103)

Utilizator DariaClemClem Daria DariaClem Data 17 mai 2022 22:17:15
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.78 kb
#include <bits/stdc++.h>

using namespace std;

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

int numere[100001], arbore[400004], nrNumere, nrOperatii;

int mijloc(int stanga, int dreapta) {
    return stanga + (dreapta - stanga) / 2;
}

//Construiesc arborele de intervale
void construireArbore(int stanga, int dreapta, int index) {
    if (stanga == dreapta)  //daca am ajuns pe o "frunza" in vectorul arbore memorez elementul din vectorul de numere
        arbore[index] = numere[stanga];
    else {
        int mij = mijloc(stanga, dreapta);
        construireArbore(stanga, mij, 2 * index);   //parcurg subarborele stang
        construireArbore(mij + 1, dreapta, 2 * index + 1);  //parcurg subarborele drept
        arbore[index] = arbore[index * 2] + arbore[index * 2 + 1];  //pentru nodul curent memorez suma fiilor sai
    }
}

//Determin suma din intervalul dat
int determinareSuma(int a, int b, int stanga, int dreapta, int index) {
    //daca intervalul curent este continut de intervalul pentru care cautam valoarea maxima returnez valoarea nodului
    if (a <= stanga and dreapta <= b) {
        return arbore[index];
    } else {
        //daca intervalele se intersecteaza are rost sa caut suma
        if (a <= dreapta and b >= stanga) {
            int mij = mijloc(stanga, dreapta);
            return determinareSuma(a, b, stanga, mij, 2 * index) +
                   determinareSuma(a, b, mij + 1, dreapta, 2 * index + 1);
        }
    }
    return 0;
}

//Functia de actualizare a arborelui
void actualizare(int stanga, int dreapta, int indexArbore, int indexSchimbat) {
    //Daca am ajuns pe o frunza in vectorul arbore actualizez valoarea acestuia
    if (stanga == dreapta) {
        arbore[indexArbore] -= numere[indexSchimbat];
    } else {
        int mij = mijloc(stanga, dreapta);
        //Daca indexul valorii schimbate e mai mica sau egala cu mijlocul inseamna ca trebuie sa actualizez valorile in subarborele stang, daca nu, in cel drept
        if (indexSchimbat <= mij)
            actualizare(stanga, mij, indexArbore * 2, indexSchimbat);
        else
            actualizare(mij + 1, dreapta, indexArbore * 2 + 1, indexSchimbat);
        arbore[indexArbore] = arbore[indexArbore * 2] + arbore[indexArbore * 2 + 1];
    }
}

int main() {
    int index, operatie, a, b, stanga = 1, dreapta;
    fin >> nrNumere >> nrOperatii;
    dreapta = nrNumere;
    for (index = 1; index <= nrNumere; index += 1) {
        fin >> numere[index];
    }
    construireArbore(stanga, dreapta, 1);
    for (index = 1; index <= nrOperatii; index += 1) {
        fin >> operatie >> a >> b;
        if (operatie == 1) {
            fout << determinareSuma(a, b, stanga, dreapta, 1) << "\n";
        } else {
            numere[a] -= b;
            actualizare(stanga, dreapta, 1, a);
        }
    }
    return 0;
}