Cod sursa(job #2564779)

Utilizator vlad082002Ciocoiu Vlad vlad082002 Data 2 martie 2020 10:20:46
Problema Arbori indexati binar Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.82 kb
#include <fstream>
using namespace std;

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

int n, m, aib[100005];

void citire() {
    fin >> n >> m;
    int x;
    for(int i = 1; i <= n; i++) {
        fin >> x;
        update(x, i);
    }
}

void update(int x, int poz) {
    for(int i = poz; i <= n; i = i&(-i))
        aib[i] += x;
}

int query(int poz) {
    int res = 0;
    for(int i = poz; i > 0; i -= i&(-i))
        res += aib[i];
    return res;
}

int main() {
    citire();
    int t, a, b
    while(m--) {
        fin >> t;
        if(t == 2) {
            fin >> a;
            fout << 1 << '\n';
        } else {
            fin >> a >> b;
            if(t == 0)
                update(b, a);
            else
                fout << query(b)-query(a-1) << '\n';
        }
    }
}