Cod sursa(job #1782289)

Utilizator AndreiFlorescuAndrei Florescu AndreiFlorescu Data 17 octombrie 2016 22:08:50
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.09 kb
#include <iostream>
#include <fstream>

#define zeros(x) ((x ^ (x - 1)) & x)

using namespace std;

int n;
int aib[15002];

void initAib (int poz, int val) {
    int i;
    for (i = poz; i <= n; i += zeros(i)) {
        aib[i] += val;
    }
}

void addAib (int poz, int val) {
    int i;
    for (i = poz; i <= n; i += zeros(i)) {
        aib[i] -= val;
    }
}

int computeAib (int poz) {
    int sol = 0, i;
    for (i = poz; i > 0; i -= zeros(i)) {
        sol += aib[i];
    }
    return sol;
}

int main() {
    ifstream file_in ("datorii.in");
    ofstream file_out ("datorii.out");

    int k;
    int poz, val, st, dr;
    int i;

    file_in >> n >> k;
    for (i = 1; i <= n; i++) {
        file_in >> val;
        initAib (i, val);
    }

    bool op;
    for (i = 0; i < k; i++) {
        file_in >> op;
        if (op) {
            file_in >> st >> dr;
            file_out << computeAib(dr) - computeAib(st - 1) << "\n";
        } else {
            file_in >> poz >> val;
            addAib (poz, val);
        }
    }

    return 0;
}