Cod sursa(job #3322035)

Utilizator AndreiRaresAcatrini Rares Andrei AndreiRares Data 12 noiembrie 2025 12:09:07
Problema Arbori indexati binar Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.17 kb
#include <iostream>
#include <fstream>
using namespace std;

#ifdef LOCAL
#define fin cin
#define fout cout
#else
ifstream fin("aib.in");
ofstream fout("aib.out");
#endif

long long n;
long long aib[100001];

void update(long long p, long long a) {
    while (p <= n) {
        aib[p] += a;
        p += (p & -p);
    }
}

long long query(long long p) {
    long long s = 0;
    while (p > 0) {
        s += aib[p];
        p -= (p & -p);
    }
    return s;
}

int main() {
    long long m, t, a, b;
    bool gasit;
    fin >> n >> m;
    for (long long i=1; i<=n; i++) {
        fin >> a;
        update(i, a);
    }
    for (long long i=1; i<=m; i++) {
        fin >> t >> a;
        if (t == 0) {
            fin >> b;
            update(a, b);
        }
        else if (t == 1) {
            fin >> b;
            fout << query(b) - query(a - 1) << '\n';
        }
        else {
            gasit = false;
            for (long long i=1; i<=n; i*=2) if (aib[i] == a) {
                fout << i << '\n';
                gasit = true;
                break;
            }
            if (!gasit) fout << "-1\n";
        }
    }
}