Cod sursa(job #3132390)

Utilizator Nicoleta114Caramaliu Nicoleta Nicoleta114 Data 22 mai 2023 16:56:23
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.93 kb
#include <fstream>
#include <vector>

using namespace std;

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

int n, m;
vector<int> valori;
vector<int> prefixSuma;

void Update(int poz, int val) {
    valori[poz] += val;
    for (int i = poz; i <= n; i += i & -i) {
        prefixSuma[i] += val;
    }
}

int getSuma(int poz) {
    int suma = 0;
    for (int i = poz; i > 0; i -= i & -i) {
        suma += prefixSuma[i];
    }
    return suma;
}

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

    valori.resize(n + 1);
    prefixSuma.resize(n + 1);

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

    for (int i = 1; i <= m; i++) {
        int t;
        f >> t;
        if (!t) {
            int val, poz;
            f >> poz >> val;
            Update(poz, -val);
        } else {
            int p, q;
            f >> p >> q;
            g << getSuma(q) - getSuma(p - 1) << '\n';
        }
    }

    return 0;
}